PR comment

openclimatefix · Sep 13, 2024 · a1b10d7 · a1b10d7
1 parent 5cfa4d7
commit a1b10d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ocf_datapipes/transform/xarray/normalize.py b/ocf_datapipes/transform/xarray/normalize.py
@@ -22,8 +22,8 @@ def __init__(
         max_value: Optional[Union[int, float]] = None,
         calculate_mean_std_from_example: bool = False,
         normalize_fn: Optional[Callable] = None,
-        min_values: Optional[np.ndarray] = None,
-        max_values: Optional[np.ndarray] = None,
+        min_values: Optional[Union[xr.Dataset, xr.DataArray, np.ndarray]] = None,
+        max_values: Optional[Union[xr.Dataset, xr.DataArray, np.ndarray]] = None,
     ):
         """
         Normalize the data with either given mean/std,