Use significantly larger cluster for small workloads

coiled · Dec 10, 2024 · a41ae99 · a41ae99
1 parent 9cbf5f6
commit a41ae99
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 8 deletions.
diff --git a/tests/geospatial/test_atmospheric_circulation.py b/tests/geospatial/test_atmospheric_circulation.py
@@ -12,7 +12,7 @@ def test_atmospheric_circulation(
         "region": "us-central1",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 100},
         "large": {"n_workers": 100},
     },

diff --git a/tests/geospatial/test_climatology.py b/tests/geospatial/test_climatology.py
@@ -23,7 +23,7 @@ def test_rechunk_map_blocks(
         "region": "us-central1",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 100},
         "large": {"n_workers": 100},
     },
@@ -49,7 +49,7 @@ def test_highlevel_api(
         "idle_timeout": "1h",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 100},
         "large": {"n_workers": 100},
     },

diff --git a/tests/geospatial/test_cloud_optimize.py b/tests/geospatial/test_cloud_optimize.py
@@ -11,7 +11,7 @@ def test_cloud_optimize(
         "region": "us-west-2",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 100},
         "large": {"n_workers": 200},
     },

diff --git a/tests/geospatial/test_rechunking.py b/tests/geospatial/test_rechunking.py
@@ -12,7 +12,7 @@ def test_era5_rechunking(
         "region": "us-central1",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 100},
         "large": {"n_workers": 100},
     },

diff --git a/tests/geospatial/test_regridding.py b/tests/geospatial/test_regridding.py
@@ -13,7 +13,7 @@ def test_xesmf(
         "wait_for_workers": True,
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "medium": {"n_workers": 10},
         "large": {"n_workers": 10},
     },

diff --git a/tests/geospatial/test_satellite_filtering.py b/tests/geospatial/test_satellite_filtering.py
@@ -14,7 +14,7 @@ def test_satellite_filtering(
         "region": "westeurope",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "large": {"n_workers": 100},
     },
 ):

diff --git a/tests/geospatial/test_zonal_average.py b/tests/geospatial/test_zonal_average.py
@@ -16,7 +16,7 @@ def test_nwm(
         "region": "us-east-1",
     },
     scale_kwargs={
-        "small": {"n_workers": 10},
+        "small": {"n_workers": 50},
         "large": {"n_workers": 200, "scheduler_memory": "32 GiB"},
     },
 ):