update

ecmwf-lab · Sep 20, 2023 · 2283ad2 · 2283ad2
1 parent fb27211
commit 2283ad2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ecml_tools/data.py b/ecml_tools/data.py
@@ -102,6 +102,10 @@ def __getitem__(self, n):
 
 class Subset(Base):
     def __init__(self, dataset, indices):
+        while isinstance(dataset, Subset):
+            indices = [dataset.indices[i] for i in indices]
+            dataset = dataset.dataset
+
         self.dataset = dataset
         self.indices = list(indices)