superb-ai-onprem 0.5.1__tar.gz → 0.5.3__tar.gz

{superb_ai_onprem-0.5.1/superb_ai_onprem.egg-info → superb_ai_onprem-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.5.1
+Version: 0.5.3
 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.1 → superb_ai_onprem-0.5.3}/spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.1'
-__version_tuple__ = version_tuple = (0, 5, 1)
+__version__ = version = '0.5.3'
+__version_tuple__ = version_tuple = (0, 5, 3)
 
-__commit_id__ = commit_id = 'gba8ec8041'
+__commit_id__ = commit_id = 'g42e97052f'

{superb_ai_onprem-0.5.1 → superb_ai_onprem-0.5.3}/spb_onprem/data/service.py

@@ -27,6 +27,7 @@ from .entities import (
     Annotation,
     DataMeta,
     Prediction,
+    Frame,
 )
 from .enums import (
     DataType,

{superb_ai_onprem-0.5.1 → superb_ai_onprem-0.5.3}/spb_onprem/slices/params/__init__.py

@@ -12,6 +12,9 @@ from .create_slice import (
 from .update_slice import (
     update_slice_params,
 )
+from .delete_slice import (
+    delete_slice_params,
+)
 
 __all__ = (
     "SlicesFilter",
@@ -20,4 +23,5 @@ __all__ = (
     "slice_params",
     "create_slice_params",
     "update_slice_params",
+    "delete_slice_params",
 )

superb_ai_onprem-0.5.3/spb_onprem/slices/params/delete_slice.py

@@ -0,0 +1,25 @@
+from spb_onprem.exceptions import BadParameterError
+
+
+def delete_slice_params(
+    dataset_id: str,
+    slice_id: str,
+):
+    """Create parameters for slice deletion.
+    
+    Args:
+        dataset_id (str): The dataset ID the slice belongs to.
+        slice_id (str): The ID of the slice to delete.
+        
+    Returns:
+        dict: Parameters for slice deletion
+    """
+    if dataset_id is None:
+        raise BadParameterError("Dataset ID is required")
+    if slice_id is None:
+        raise BadParameterError("Slice ID is required")
+
+    return {
+        "dataset_id": dataset_id,
+        "id": slice_id,
+    }

{superb_ai_onprem-0.5.1 → superb_ai_onprem-0.5.3}/spb_onprem/slices/queries.py

@@ -3,6 +3,7 @@ from spb_onprem.slices.params import (
     slice_params,
     create_slice_params,
     update_slice_params,
+    delete_slice_params,
 )
 
 
@@ -119,3 +120,16 @@ class Queries:
         ''',
         "variables": update_slice_params,
     }
+
+    DELETE_SLICE = {
+        "name": "deleteSlice",
+        "query": '''
+            mutation deleteSlice(
+                $dataset_id: String!,
+                $id: ID!
+            ) {
+                deleteSlice(datasetId: $dataset_id, id: $id)
+            }
+        ''',
+        "variables": delete_slice_params,
+    }

{superb_ai_onprem-0.5.1 → superb_ai_onprem-0.5.3}/spb_onprem/slices/service.py

@@ -171,3 +171,26 @@ class SliceService(BaseService):
         )
         slice_dict = response.get("updateSlice", {})
         return Slice.model_validate(slice_dict)
+
+    def delete_slice(
+        self,
+        dataset_id: str,
+        slice_id: str,
+    ) -> bool:
+        """Delete a slice.
+        
+        Args:
+            dataset_id (str): The ID of the dataset the slice belongs to.
+            slice_id (str): The ID of the slice to delete.
+        
+        Returns:
+            bool: True if the slice was deleted, False otherwise.
+        """
+        response = self.request_gql(
+            Queries.DELETE_SLICE,
+            Queries.DELETE_SLICE["variables"](
+                dataset_id=dataset_id,
+                slice_id=slice_id,
+            )
+        )
+        return response

{superb_ai_onprem-0.5.1 → superb_ai_onprem-0.5.3/superb_ai_onprem.egg-info}/PKG-INFO

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