Fix usage of wait_for_workers

coiled · Oct 23, 2024 · c972fef · c972fef
1 parent bf231bb
commit c972fef
Show file tree

Hide file tree

Showing 7 changed files with 2 additions and 7 deletions.
diff --git a/tests/geospatial/conftest.py b/tests/geospatial/conftest.py
@@ -41,6 +41,8 @@ def _(n_workers, env=None, **cluster_kwargs):
             if env:
                 cluster.send_private_envs(env=env)
             with cluster.get_client() as client:
+                # FIXME https://github.com/coiled/platform/issues/103
+                client.wait_for_workers(n_workers)
                 with benchmark_all(client):
                     yield client
 

diff --git a/tests/geospatial/test_atmospheric_circulation.py b/tests/geospatial/test_atmospheric_circulation.py
@@ -9,7 +9,6 @@ def test_atmospheric_circulation(
     cluster_kwargs={
         "workspace": "dask-benchmarks-gcp",
         "region": "us-central1",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},

diff --git a/tests/geospatial/test_climatology.py b/tests/geospatial/test_climatology.py
@@ -80,7 +80,6 @@ def test_rechunk_map_blocks(
     cluster_kwargs={
         "workspace": "dask-benchmarks-gcp",
         "region": "us-central1",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},
@@ -143,7 +142,6 @@ def test_highlevel_api(
     cluster_kwargs={
         "workspace": "dask-benchmarks-gcp",
         "region": "us-central1",
-        "wait_for_workers": True,
         "idle_timeout": "1h",
     },
     scale_kwargs={

diff --git a/tests/geospatial/test_cloud_optimize.py b/tests/geospatial/test_cloud_optimize.py
@@ -9,7 +9,6 @@ def test_cloud_optimize(
     cluster_kwargs={
         "workspace": "dask-benchmarks",
         "region": "us-west-2",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},

diff --git a/tests/geospatial/test_rechunking.py b/tests/geospatial/test_rechunking.py
@@ -9,7 +9,6 @@ def test_era5_rechunking(
     cluster_kwargs={
         "workspace": "dask-benchmarks-gcp",
         "region": "us-central1",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},

diff --git a/tests/geospatial/test_satellite_filtering.py b/tests/geospatial/test_satellite_filtering.py
@@ -61,7 +61,6 @@ def test_satellite_filtering(
     cluster_kwargs={
         "workspace": "dask-benchmarks-azure",
         "region": "westeurope",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},

diff --git a/tests/geospatial/test_zonal_average.py b/tests/geospatial/test_zonal_average.py
@@ -15,7 +15,6 @@ def test_nwm(
     cluster_kwargs={
         "workspace": "dask-benchmarks",
         "region": "us-east-1",
-        "wait_for_workers": True,
     },
     scale_kwargs={
         "small": {"n_workers": 10},