dim instead of dims

Author: FynnBe

bioimageio/core/prediction_pipeline/_combined_processing.py

@@ -155,9 +155,9 @@ def _compute_tensor_statistics(tensor: xr.DataArray, measures: Set[Measure]) ->
         ret = {}
         for measure in measures:
             if isinstance(measure, Mean):
-                v = tensor.mean(dims=measure.axes)
+                v = tensor.mean(dim=measure.axes)
             elif isinstance(measure, Std):
-                v = tensor.std(dims=measure.axes)
+                v = tensor.std(dim=measure.axes)
             elif isinstance(measure, Percentile):
                 v = tensor.quantile(measure.n / 100.0, dim=measure.axes)
             else: