scale-nucleus 0.17.6b3__tar.gz → 0.17.7__tar.gz

{scale_nucleus-0.17.6b3 → scale_nucleus-0.17.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scale-nucleus
-Version: 0.17.6b3
+Version: 0.17.7
 Summary: The official Python client library for Nucleus, the Data Platform for AI
 Home-page: https://scale.com/nucleus
 License: MIT
@@ -14,7 +14,6 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: launch
 Provides-Extra: metrics
 Requires-Dist: Pillow (>=7.1.2)

{scale_nucleus-0.17.6b3 → scale_nucleus-0.17.7}/nucleus/constants.py

@@ -114,8 +114,7 @@ MODEL_PREDICTION_ID_KEY = "model_prediction_id"
 MODEL_PREDICTION_LABEL_KEY = "model_prediction_label"
 NAME_KEY = "name"
 NEW_ITEMS = "new_items"
-# NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
-NUCLEUS_ENDPOINT = "http://localhost:3003/nucleus"
+NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
 NUM_SENSORS_KEY = "num_sensors"
 ORIGINAL_IMAGE_URL_KEY = "original_image_url"
 PAGE_SIZE_KEY = "pageSize"

{scale_nucleus-0.17.6b3 → scale_nucleus-0.17.7}/nucleus/dataset.py

@@ -1449,10 +1449,13 @@ class Dataset:
         )
         return convert_export_payload(api_payload[EXPORTED_ROWS])
 
-    def scene_and_annotation_generator(self, page_size: int = 10):
+    def scene_and_annotation_generator(
+        self, slice_id=None, page_size: int = 10
+    ):
         """Provides a generator of all Scenes and Annotations in the dataset grouped by scene.
 
         Args:
+            slice_id: 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 @@ class Dataset:
             endpoint=f"dataset/{self.id}/{endpoint_name}",
             result_key=EXPORT_FOR_TRAINING_KEY,
             page_size=page_size,
+            sliceId=slice_id,
         )
 
         for data in json_generator:

{scale_nucleus-0.17.6b3 → scale_nucleus-0.17.7}/pyproject.toml

@@ -25,7 +25,7 @@ ignore = ["E501", "E741", "E731", "F401"]  # Easy ignore for getting it running
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.17.6b3"
+version = "0.17.7"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]