Merge pull request #104 from hoetmaaiers/master

Allow ResultSet slicing
ESGF · Nov 25, 2024 · 848f09b · 848f09b
2 parents cd65b58 + 88712bc
commit 848f09b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyesgf/search/results.py b/pyesgf/search/results.py
@@ -42,6 +42,10 @@ def __init__(self, context, batch_size=DEFAULT_BATCH_SIZE, eager=True):
             self.__get_batch(0)
 
     def __getitem__(self, index):
+        if isinstance(index, slice):
+            # Handle slicing
+            return [self[i] for i in range(*index.indices(len(self)))]
+
         batch_i = index // self.batch_size
         offset = index % self.batch_size
         batch = self.__get_batch(batch_i)