superb-ai-onprem 0.5.8__tar.gz → 0.5.9__tar.gz

{superb_ai_onprem-0.5.8/superb_ai_onprem.egg-info → superb_ai_onprem-0.5.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.8
+Version: 0.5.9
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.9}/spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.8'
-__version_tuple__ = version_tuple = (0, 5, 8)
+__version__ = version = '0.5.9'
+__version_tuple__ = version_tuple = (0, 5, 9)
 
-__commit_id__ = commit_id = 'g07632cdb4'
+__commit_id__ = commit_id = 'gd78508cdd'

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.9}/spb_onprem/data/service.py

@@ -251,25 +251,28 @@ class DataService(BaseService):
                 key,
             )
 
+        # Create Data object for the create_params function
+        data_obj = Data(
+            dataset_id=dataset_id,
+            key=key,
+            type=DataType.SUPERB_IMAGE,
+            slice_ids=slices,
+            scene=[Scene(
+                id=f"{key}_scene_0",
+                type=SceneType.IMAGE,
+                content=content,
+                meta={}
+            )],
+            thumbnail=content,
+            annotation=annotation,
+            predictions=predictions,
+            meta=meta,
+            system_meta=system_meta,
+        )
+
         response = self.request_gql(
             Queries.CREATE,
-            Queries.CREATE["variables"](
-                dataset_id=dataset_id,
-                key=key,
-                type=DataType.SUPERB_IMAGE,
-                slices=slices,
-                scene=[{
-                    "id": f"{key}_scene_0",
-                    "type": SceneType.IMAGE,
-                    "content": content.model_dump(by_alias=True),
-                    "meta": {}
-                }],
-                thumbnail=content.model_dump(by_alias=True),
-                annotation=annotation,
-                predictions=predictions,
-                meta=meta,
-                system_meta=system_meta,
-            )
+            Queries.CREATE["variables"](data_obj)
         )
         data = Data.model_validate(response)
         return data

{superb_ai_onprem-0.5.8 → superb_ai_onprem-0.5.9/superb_ai_onprem.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.8
+Version: 0.5.9
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI