Try dask submit/result for Windows

EcoExtreML · Jul 26, 2024 · 1fe26d0 · 1fe26d0
1 parent aa650d3
commit 1fe26d0
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/tests/test_datasets/test_fapar_lai.py b/tests/test_datasets/test_fapar_lai.py
@@ -87,15 +87,19 @@ def test_download(self, mock_retrieve, valid_path_config, dummy_dir):
     @pytest.mark.slow
     def test_ingest(self, dummy_dir):
         """Test ingest function."""
-        dask.distributed.Client()
+        c = dask.distributed.Client()
 
         ingest_dir = Path(dummy_dir) / "ingest"
         ingest_dir.mkdir()
 
-        lai_dataset = FaparLAI()
-        lai_dataset.ingest(
-            download_dir=data_folder / "fapar-lai" / "download", ingest_dir=ingest_dir
-        )
+        def ingest_lai():
+            lai_dataset = FaparLAI()
+            lai_dataset.ingest(
+                download_dir=data_folder / "fapar-lai" / "download", ingest_dir=ingest_dir
+            )
+        f = c.submit(ingest_lai)
+        f.result()
+
         ds = xr.open_mfdataset((ingest_dir / "fapar-lai").glob("*.nc"))
         assert isinstance(ds, xr.Dataset)