From e9c0701bce9e5dfff7184fc6210092b82b9386a6 Mon Sep 17 00:00:00 2001 From: Ana Prieto Nemesio <91897203+anaprietonem@users.noreply.github.com> Date: Fri, 28 Feb 2025 10:20:52 +0100 Subject: [PATCH] fix: datashader new release test (#104) * datashader new release test * remove for py3.9 compatibility --- training/pyproject.toml | 1 - training/src/anemoi/training/diagnostics/callbacks/plot.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/training/pyproject.toml b/training/pyproject.toml index ec1eaf3f..2613f37e 100644 --- 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 index 38822de0..9aa6fb4c 100644 --- 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):