more doc

ecmwf · Mar 28, 2024 · d5c1a77 · d5c1a77
1 parent 7c24b8f
commit d5c1a77
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/anemoi/datasets/data/dataset.py b/anemoi/datasets/data/dataset.py
@@ -65,10 +65,12 @@ def _subset(self, **kwargs):
             return Rename(self, rename)._subset(**kwargs)
 
         if "statistics" in kwargs:
+            from ..data import open_dataset
             from .statistics import Statistics
 
             statistics = kwargs.pop("statistics")
-            return Statistics(self, statistics)._subset(**kwargs)
+
+            return Statistics(self, open_dataset(statistics))._subset(**kwargs)
 
         if "thinning" in kwargs:
             from .masked import Thinning

diff --git a/anemoi/datasets/data/masked.py b/anemoi/datasets/data/masked.py
@@ -86,7 +86,9 @@ def tree(self):
 
 class Cropping(Masked):
     def __init__(self, forward, area):
-        self.area = area
+        from ..data import open_dataset
+
+        area = area if isinstance(area, (list, tuple)) else open_dataset(area)
 
         if isinstance(area, Dataset):
             north = np.amax(area.latitudes)
@@ -95,6 +97,7 @@ def __init__(self, forward, area):
             west = np.amin(area.longitudes)
             area = (north, west, south, east)
 
+        self.area = area
         mask = cropping_mask(forward.latitudes, forward.longitudes, *area)
 
         super().__init__(forward, mask)