xarray 2023 compatibility

Author: FynnBe

bioimageio/core/stat_calculators.py

@@ -139,6 +139,11 @@ def compute(
         else:
             n = int(np.prod([tensor.sizes[d] for d in self._axes]))
 
+        if xr.__version__.startswith("2023"):
+            var = xr.dot(c, c, dims=self._axes) / n
+        else:
+            var = xr.dot(c, c, dim=self._axes) / n
+
         var = xr.dot(c, c, dim=self._axes) / n
         assert isinstance(var, xr.DataArray)
         std = np.sqrt(var)

dev/env-py38.yaml

@@ -39,6 +39,6 @@ dependencies:
   - torchvision
   - tqdm
   - typing-extensions
-  - xarray>=2024.01,<2025.3.0
+  - xarray>=2023.01,<2025.3.0
   - pip:
       - -e ..

setup.py

@@ -42,7 +42,7 @@
         "ruyaml",
         "tqdm",
         "typing-extensions",
-        "xarray>=2024.01,<2025.3.0",
+        "xarray>=2023.01,<2025.3.0",
     ],
     include_package_data=True,
     extras_require={