From 3325284b8f17523551a5a38df41a733eb2d46449 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Fri, 20 Sep 2024 12:28:46 -0500 Subject: [PATCH] Update --- .github/workflows/tests.yml | 2 +- tests/geospatial/test_atmospheric_circulation.py | 15 +++++++-------- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 9fed28eafd..69d483925d 100644 --- 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 index 3e8e257c9f..b70885d19c 100644 --- 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()})