optional slice param (#441)

Author: iz18

nucleus/dataset.py

@@ -1449,10 +1449,13 @@ def items_and_annotations(
         )
         return convert_export_payload(api_payload[EXPORTED_ROWS])
 
-    def scene_and_annotation_generator(self, page_size: int = 10):
+    def scene_and_annotation_generator(
+        self, sliceId=None, page_size: int = 10
+    ):
         """Provides a generator of all Scenes and Annotations in the dataset grouped by scene.
 
         Args:
+            sliceId: Optional slice ID to filter the scenes and annotations.
             page_size: Number of scenes to fetch per page. Default is 10.
 
         Returns:
@@ -1505,6 +1508,7 @@ def scene_and_annotation_generator(self, page_size: int = 10):
             endpoint=f"dataset/{self.id}/{endpoint_name}",
             result_key=EXPORT_FOR_TRAINING_KEY,
             page_size=page_size,
+            sliceId=sliceId,
         )
 
         for data in json_generator: