scale-nucleus 0.17.6__tar.gz → 0.17.6b1__tar.gz

{scale_nucleus-0.17.6 → scale_nucleus-0.17.6b1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scale-nucleus
-Version: 0.17.6
+Version: 0.17.6b1
 Summary: The official Python client library for Nucleus, the Data Platform for AI
 Home-page: https://scale.com/nucleus
 License: MIT
@@ -14,6 +14,7 @@ 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.6 → scale_nucleus-0.17.6b1}/nucleus/constants.py

@@ -114,7 +114,8 @@ 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 = "https://api.scale.com/v1/nucleus"
+NUCLEUS_ENDPOINT = "localhost:3003/nucleus"
 NUM_SENSORS_KEY = "num_sensors"
 ORIGINAL_IMAGE_URL_KEY = "original_image_url"
 PAGE_SIZE_KEY = "pageSize"

{scale_nucleus-0.17.6 → scale_nucleus-0.17.6b1}/nucleus/dataset.py

@@ -997,9 +997,9 @@ class Dataset:
             raise Exception("Must provide at least one list of internal IDs")
         payload = {
             NAME_KEY: name,
-            DATASET_ITEM_IDS_KEY: dataset_item_ids
-            if dataset_item_ids
-            else None,
+            DATASET_ITEM_IDS_KEY: (
+                dataset_item_ids if dataset_item_ids else None
+            ),
             SCENE_IDS_KEY: scene_ids if scene_ids else None,
             OBJECT_IDS_KEY: [*(annotation_ids or []), *(prediction_ids or [])]
             or None,
@@ -1449,16 +1449,34 @@ class Dataset:
         )
         return convert_export_payload(api_payload[EXPORTED_ROWS])
 
-    def scene_and_annotation_generator(self, page_size=10):
-        """Provides a generator of all DatasetItems and Annotations in the dataset grouped by scene.
+    def scene_and_annotation_generator(self, page_size: int = 10):
+        """Provides a generator of all Scenes and Annotations in the dataset grouped by scene.
 
+        Args:
+            page_size: Number of scenes to fetch per page. Default is 10.
 
         Returns:
-            Generator where each element is a nested dict (representing a JSON) structured in the following way:
+            Generator where each element is a nested dict containing scene and annotation information of the dataset structured as a JSON.
+
+            Track grouping is slightly more complicated and is done in the following order:
+            1. If track_id is defined in the annotations table, use it as the track id.
+            2. If track_reference_id is defined in the metadata field of the annotations table, use it as the track id.
+            3. If track_id is defined in the metadata field of the annotations table, use it as the track id.
+            4. If track_id is not defined and annotation_id is defined, use annotation_id as track id.
+            5. If annotation_id grouping is unsuccessful such that the annotation id is unique across all frames in the scene for all annotations, throw an error that the annotation format is incompatible.
+            6. If there is no track or annotation id, throw an error that the annotation format is incompatible.
+
+            If you use the generator and discover that no scenes were generated, check the response error message for more information. It is likely that the annotations are not in the correct format.
 
+            ::
             Iterable[{
-                "file_location": str,
-                "metadata": Dict[str, Any],
+                "scene": {
+                    "id": str,
+                    "reference_id": str,
+                    "metadata": Dict[str, Any]
+                    "type": str,
+                    "fileLocation": str,
+                }
                 "annotations": {
                     "{trackId}": {
                         "label": str,

{scale_nucleus-0.17.6 → scale_nucleus-0.17.6b1}/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.6"
+version = "0.17.6b1"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]