Configure A/B test

coiled · Sep 13, 2024 · bab0e14 · bab0e14
1 parent 75a4c91
commit bab0e14
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 9 deletions.
diff --git a/AB_environments/AB_baseline.conda.yaml b/AB_environments/AB_baseline.conda.yaml
@@ -52,6 +52,6 @@ dependencies:
     # conda. You may alternatively point to your own git fork (but make sure you
     # sync'ed tags!)
     # Read README.md for troubleshooting.
-    # - git+https://github.com/dask/dask@191d39177009d2cce25b818878118e35329b6db3
-    # - git+https://github.com/dask/distributed@0304fb6e665e36abf9e3086173cccd36e29ae84d
-    # - git+https://github.com/dask-contrib/dask-expr@9f765764da3f518ddd4c896c98b8a40a979a5553
+    - git+https://github.com/dask/dask@89189546ac7f2695b34edf2318ab65f693343da7
+    - git+https://github.com/dask/distributed@ec3f4eccd1b8f8541dfa3b72e363464192823a4a
+    - git+https://github.com/dask-contrib/dask-expr@92e789086cc7e8b2c3ae670b2b423925062d7c2e
diff --git a/AB_environments/AB_sample.conda.yaml b/AB_environments/AB_sample.conda.yaml
@@ -58,6 +58,6 @@ dependencies:
     # conda. You may alternatively point to your own git fork (but make sure you
     # sync'ed tags!)
     # Read README.md for troubleshooting.
-    - git+https://github.com/dask/dask@191d39177009d2cce25b818878118e35329b6db3
-    - git+https://github.com/dask/distributed@0304fb6e665e36abf9e3086173cccd36e29ae84d
-    - git+https://github.com/dask-contrib/dask-expr@9f765764da3f518ddd4c896c98b8a40a979a5553
+    - git+https://github.com/dask/dask@89189546ac7f2695b34edf2318ab65f693343da7
+    - git+https://github.com/dask/distributed@1c0a178134c06aca977d0dadd2572e2f3cf029e4
+    - git+https://github.com/dask-contrib/dask-expr@92e789086cc7e8b2c3ae670b2b423925062d7c2e
diff --git a/AB_environments/config.yaml b/AB_environments/config.yaml
@@ -3,7 +3,7 @@
 # Setting it to 5 is a good value to get statistically significant results.
 # This must remain set to 0 in the main branch, thus completely disabling
 # A/B tests, in order to avoid unnecessary runs.
-repeat: 0
+repeat: 5
 
 # Set to true to automatically create a verbatim copy of AB_baseline and then compare
 # the two in the A/B tests. Set to false to save some money if you are already confident
@@ -13,7 +13,13 @@ test_null_hypothesis: true
 # Test directories, test files, or individual tests to run.
 targets:
   # - tests
-  - tests/benchmarks
+  - tests/benchmarks/test_array.py
+  - tests/benchmarks/test_h2o.py
+  - tests/benchmarks/test_join.py
+  - tests/benchmarks/test_rechunk.py
+  - tests/benchmarks/test_xarray.py
+  - tests/benchmarks/test_zarr.py
+  - tests/tpch
   # - tests/runtime
   # - tests/stability
   # - tests/tpch/test_dask.py

diff --git a/tests/geospatial/test_rechunking.py b/tests/geospatial/test_rechunking.py
@@ -13,7 +13,7 @@ def test_era5_rechunking(client, gcs_url, scale):
     if scale == "small":
         # 101.83 GiB (small)
         time_range = slice("2020-01-01", "2023-01-01")
-        variables = ["sea_surface_temperature"]
+        variables = ["sea_surface_temperature", "snow_depth"]
     elif scale == "medium":
         # 2.12 TiB (medium)
         time_range = slice(None)