remove upper xarray pin

Author: FynnBe

changelog.md

@@ -3,6 +3,7 @@
 - bump bioimageio.spec library version to 0.5.5.6
 - Replace `conda activate <env name>` with `conda run -n <env name> python --version` when checking if a conda environment exists
   (This is closer to the actual `conda run` command we need and avoids requests by conda to rerun `conda init` (in CI).)
+- remove upper xarray pin (use ops from xarray.computation.ops, fallback to xarray.core.ops for older xarray versions)
 
 ### 0.9.3
 

pyproject.toml

@@ -18,7 +18,7 @@ dependencies = [
     "scipy",
     "tqdm",
     "typing-extensions",
-    "xarray>=2023.01,<2025.3.0",
+    "xarray>=2023.01",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

src/bioimageio/core/_magic_tensor_ops.py

@@ -6,7 +6,14 @@
 from typing import Any, Callable
 
 from typing_extensions import Self
-from xarray.core import nputils, ops
+from xarray.core import nputils
+
+try:
+    # xarray >= 2025.03
+    from xarray.computation import ops
+except ImportError:
+    # xarray < 2025.03
+    from xarray.core import ops  # type: ignore
 
 
 class MagicTensorOpsMixin: