Skip to content

Commit 6a06869

Browse files
authored
Added coiled cluster config.
1 parent c63d25e commit 6a06869

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

demo/sst

+11-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,16 @@ gcloud auth application-default login
99
import xarray as xr
1010
import xarray_sql as qr
1111

12-
# TODO(alxmrs): Add coiled or dask cluster code.
12+
from coiled import Cluster
13+
14+
cluster = Cluster(
15+
region='us-central1',
16+
worker_memory='16 GiB',
17+
spot_policy='spot_with_fallback',
18+
arm=True,
19+
)
20+
client = cluster.get_client()
21+
cluster.adapt(minimum=1, maximum=100)
1322

1423
era5_ds = xr.open_zarr(
1524
'gs://gcp-public-data-arco-era5/ar/'
@@ -22,7 +31,7 @@ era5_sst_ds = era5_ds[['sea_surface_temperature']].sel(
2231
level=1000, # surface level only.
2332
)
2433

25-
# chunk sizes determined from VM memory limit of 16 GB.
34+
# chunk sizes determined from VM memory limit of 16 GiB.
2635
c = qr.Context()
2736
c.create_table('era5', era5_sst_ds, chunks=dict(time=24))
2837

0 commit comments

Comments
 (0)