Update

coiled · Sep 20, 2024 · 3325284 · 3325284
1 parent cb638d0
commit 3325284
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -31,7 +31,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python_version: ["3.10"]
-        pytest_args: [tests/geospatial/test_atmospheric_circulation.py --ignore=tests/tpch]
+        pytest_args: [tests --ignore=tests/tpch]
         extra-env: [""]
         name_prefix: [tests]
         include:

diff --git a/tests/geospatial/test_atmospheric_circulation.py b/tests/geospatial/test_atmospheric_circulation.py
@@ -61,14 +61,13 @@ def test_atmospheric_circulation(
         temdiags = zonal_means.merge(anomaly[["uv", "vt", "uw"]].mean("longitude"))
 
         # This is incredibly slow, takes a while for flox to construct the graph
-        # daily = temdiags.resample(time="D").mean()
+        daily = temdiags.resample(time="D").mean()
 
-        # Option 2: rechunk to make it a blockwise problem
-        # we should do this automatically
-        daily = (
-            temdiags.chunk(time=xr.groupers.TimeResampler("D"))
-            .resample(time="D")
-            .mean()
-        )
+        # # Users often rework things via a rechunk to make this a blockwise problem
+        # daily = (
+        #     temdiags.chunk(time=xr.groupers.TimeResampler("D"))
+        #     .resample(time="D")
+        #     .mean()
+        # )
 
         daily.to_zarr(gcs_url, storage_options={"token": CoiledShippedCredentials()})