fix: datashader new release test (#104)

* datashader new release test * remove for py3.9 compatibility
ecmwf · Feb 28, 2025 · e9c0701 · e9c0701
1 parent 42c5706
commit e9c0701
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/training/pyproject.toml b/training/pyproject.toml
@@ -44,7 +44,6 @@ dependencies = [
   "anemoi-graphs>=0.4.1",
   "anemoi-models>=0.4.1",
   "anemoi-utils[provenance]>=0.4.4",
-  "dask<=2024.12.1",                 # pinned until a new datashader version is released
   "datashader>=0.16.3",
   "einops>=0.6.1",
   "hydra-core>=1.3",

diff --git a/training/src/anemoi/training/diagnostics/callbacks/plot.py b/training/src/anemoi/training/diagnostics/callbacks/plot.py
@@ -804,7 +804,7 @@ def automatically_determine_group(name: str) -> str:
 
         # set x-ticks
         x_tick_positions = np.cumsum(group_counts) - group_counts / 2 - 0.5
-        xticks = dict(zip(unique_group_list, x_tick_positions, strict=False))
+        xticks = dict(zip(unique_group_list, x_tick_positions))
 
         legend_patches = []
         for group_idx, group in enumerate(unique_group_list):