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

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

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

@@ -27,6 +27,7 @@ from .entities import (
     ActivityHistory,
     Export,
     Content,
+    Frame,
 
     # Enums
     DataType,
@@ -91,6 +92,7 @@ __all__ = (
     "ActivityHistory",
     "Export",
     "Content",
+    "Frame",
     
     # Enums
     "DataType",

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/_version.py

@@ -1,7 +1,14 @@
 # file generated by setuptools-scm
 # don't change, don't track in version control
 
-__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
@@ -9,13 +16,19 @@ if TYPE_CHECKING:
     from typing import Union
 
     VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
 else:
     VERSION_TUPLE = object
+    COMMIT_ID = object
 
 version: str
 __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
 
-__version__ = version = '0.4.5'
-__version_tuple__ = version_tuple = (0, 4, 5)
+__version__ = version = '0.5.1'
+__version_tuple__ = version_tuple = (0, 5, 1)
+
+__commit_id__ = commit_id = 'gba8ec8041'

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/entities/__init__.py

@@ -4,9 +4,11 @@ from .data import Data
 from .prediction import Prediction
 from .scene import Scene
 from .data_slice import DataSlice
+from .frame import Frame
 
 
 __all__ = (
+    "Frame",
     "Data",
     "Scene",
     "Annotation",

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/entities/annotation.py

@@ -11,7 +11,7 @@ class AnnotationVersion(CustomBaseModel):
     This has the content of the data annotation.
     """
     id: Optional[str] = None
-    channel: Optional[str] = None
+    channels: Optional[List[str]] = None
     version: Optional[str] = None
     content: Optional[BaseContent] = None
     meta: Optional[dict] = None

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/entities/data.py

@@ -6,6 +6,7 @@ from .annotation import Annotation
 from .prediction import Prediction
 from .data_meta import DataMeta
 from .data_slice import DataSlice
+from .frame import Frame
 from spb_onprem.contents.entities import BaseContent
 
 class Data(CustomBaseModel):
@@ -18,6 +19,7 @@ class Data(CustomBaseModel):
     key: Optional[str] = None
     type: Optional[DataType] = None
     scene: Optional[List[Scene]] = None
+    frames: Optional[List[Frame]] = None
     thumbnail: Optional[BaseContent] = None
     annotation: Optional[Annotation] = None
     predictions: Optional[List[Prediction]] = None

superb_ai_onprem-0.5.1/spb_onprem/data/entities/frame.py

@@ -0,0 +1,19 @@
+from typing import Optional
+from spb_onprem.base_model import CustomBaseModel, Field
+
+
+class GeoLocation(CustomBaseModel):
+    lat: float
+    lon: float
+
+
+class Frame(CustomBaseModel):
+    """
+    The frame of the data.
+    Frame is the representation of a single frame in a sequence of data, such as a video or time series.
+    """
+    id: str
+    index: Optional[int] = None
+    captured_at: Optional[str] = Field(None, alias="capturedAt")
+    geo_location: Optional[GeoLocation] = Field(None, alias="geoLocation")
+    meta: Optional[dict] = None

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/params/__init__.py

@@ -63,6 +63,9 @@ from .change_data_reviewer import (
 from .update_data_slice import (
     update_data_slice_params,
 )
+from .update_frames import (
+    update_frames_params,
+)
 
 __all__ = [
     "create_params",
@@ -90,4 +93,5 @@ __all__ = [
     "change_data_labeler_params",
     "change_data_reviewer_params",
     "update_data_slice_params",
+    "update_frames_params",
 ]

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/params/insert_annotation_version.py

@@ -23,9 +23,9 @@ def insert_annotation_version_params(
         "meta": version.meta
     }
     
-    # channel과 version 필드가 있으면 추가
-    if version.channel is not None:
-        version_data["channel"] = version.channel
+    # channels과 version 필드가 있으면 추가
+    if version.channels is not None:
+        version_data["channels"] = version.channels
     if version.version is not None:
         version_data["version"] = version.version
 

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/params/insert_slice_annotation_version.py

@@ -24,10 +24,10 @@ def insert_slice_annotation_version_params(
         ),
         "meta": version.meta
     }
-    
-    # channel과 version 필드가 있으면 추가
-    if version.channel is not None:
-        version_data["channel"] = version.channel
+
+    # channels과 version 필드가 있으면 추가
+    if version.channels is not None:
+        version_data["channels"] = version.channels
     if version.version is not None:
         version_data["version"] = version.version
 

superb_ai_onprem-0.5.1/spb_onprem/data/params/update_frames.py

@@ -0,0 +1,40 @@
+import json
+from typing import Union, List
+from spb_onprem.data.entities import Frame
+from spb_onprem.base_types import UndefinedType, Undefined
+
+
+def update_frames_params(
+    dataset_id: str,
+    data_id: str,
+    frames: Union[List[Frame], UndefinedType, None] = Undefined,  
+):
+    """Update frames of selected data.
+    Args:
+        dataset_id (str): dataset id which the data belongs to
+        data_id (str): data id to be updated
+        frames (list[Frame]): list of frames to be updated  
+        
+    Returns:
+        dict: the params for graphql query
+    """
+    result = {
+        "dataset_id": dataset_id,
+        "data_id": data_id,
+    }
+    if frames is not Undefined:
+        result["frames"] = []
+        for frame in frames:
+            frame_data = {
+                "index": frame.index,
+                "capturedAt": frame.captured_at,
+                "meta": json.dumps(frame.meta) if frame.meta else None,
+            }
+            if frame.geo_location is not None:
+                frame_data["geoLocation"] = {
+                    "lat": frame.geo_location.lat,
+                    "lon": frame.geo_location.lon
+                }
+            result["frames"].append(frame_data)
+
+    return result

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/params/update_slice_annotation_version.py

@@ -1,4 +1,4 @@
-from typing import Union, Optional, Any
+from typing import Union, Optional, Any, List
 from spb_onprem.base_types import UndefinedType, Undefined
 from spb_onprem.exceptions import BadParameterError
 
@@ -8,7 +8,7 @@ def update_slice_annotation_version_params(
     data_id: str,
     slice_id: str,
     id: str,
-    channel: Union[str, UndefinedType, None] = Undefined,
+    channels: Union[List[str], UndefinedType, None] = Undefined,
     version: Union[str, UndefinedType, None] = Undefined,
     meta: Union[dict, UndefinedType, None] = Undefined,
 ):
@@ -19,7 +19,7 @@ def update_slice_annotation_version_params(
         data_id (str): The ID of the data.
         slice_id (str): The slice ID.
         id (str): The annotation version ID.
-        channel (str, optional): The channel of the annotation version.
+        channels (list[str], optional): The channels of the annotation version.
         version (str, optional): The version string of the annotation version.
         meta (dict, optional): The meta of the annotation version.
     """
@@ -30,8 +30,8 @@ def update_slice_annotation_version_params(
         "id": id,
     }
 
-    if channel is not Undefined:
-        variables["channel"] = channel
+    if channels is not Undefined:
+        variables["channels"] = channels
     if version is not Undefined:
         variables["version"] = version
     if meta is not Undefined:

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/data/queries.py

@@ -20,6 +20,7 @@ from .params import (
     change_data_labeler_params,
     change_data_reviewer_params,
     update_data_slice_params,
+    update_frames_params,
 )
 
 
@@ -46,10 +47,21 @@ class Schemas:
             }
             meta
         }
+        frames {
+            id
+            index
+            capturedAt
+            geoLocation {
+                lat
+                lon
+            }
+            meta
+        }
         annotation {
+            meta
             versions {
                 id
-                channel
+                channels
                 version
                 content {
                     id
@@ -57,23 +69,11 @@ class Schemas:
                 meta
             }
         }
-        predictions {
-            setId
-            content {
-                id
-            }
-            meta
-        }
         meta {
             key
             type
             value
         }
-        systemMeta {
-            key
-            type
-            value
-        }
         slices {
             id
             status
@@ -84,7 +84,7 @@ class Schemas:
             annotation {
                 versions {
                     id
-                    channel
+                    channels
                     version
                     content {
                         id
@@ -574,3 +574,23 @@ class Queries():
         ''',
         "variables": update_data_slice_params,
     }
+    
+    UPDATE_FRAMES = {
+        "name": "updateFrames",
+        "query": f'''
+            mutation (
+                $dataset_id: ID!,
+                $data_id: ID!,
+                $frames: [DataFrameInput!],
+            ) {{
+                updateFrames(
+                    datasetId: $dataset_id,
+                    dataId: $data_id,
+                    frames: $frames,
+                ) {{
+                    {Schemas.DATA}
+                }}
+            }}
+        ''',
+        "variables": update_frames_params,
+    }

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

@@ -572,7 +572,7 @@ class DataService(BaseService):
         data_id: str,
         slice_id: str,
         id: str,
-        channel: Union[str, UndefinedType, None] = Undefined,
+        channels: Union[List[str], UndefinedType, None] = Undefined,
         version: Union[str, UndefinedType, None] = Undefined,
         meta: Union[dict, UndefinedType, None] = Undefined,
     ):
@@ -583,7 +583,7 @@ class DataService(BaseService):
             data_id (str): The data id.
             slice_id (str): The slice id.
             id (str): The annotation version id.
-            channel (Union[str, UndefinedType, None], optional): The channel. Defaults to Undefined.
+            channels (Union[List[str], UndefinedType, None], optional): The channels. Defaults to Undefined.
             version (Union[str, UndefinedType, None], optional): The version. Defaults to Undefined.
             meta (Union[dict, UndefinedType, None], optional): The meta. Defaults to Undefined.
 
@@ -606,7 +606,7 @@ class DataService(BaseService):
                 data_id=data_id,
                 slice_id=slice_id,
                 id=id,
-                channel=channel,
+                channels=channels,
                 version=version,
                 meta=meta,
             )
@@ -804,3 +804,34 @@ class DataService(BaseService):
         )
         data = Data.model_validate(response)
         return data
+
+    def update_frames(
+        self,
+        dataset_id: str,
+        data_id: str,
+        frames: Union[List[Frame], UndefinedType, None] = Undefined,  
+    ):
+        """Update frames of selected data.
+        Args:
+            dataset_id (str): dataset id which the data belongs to
+            data_id (str): data id to be updated
+            frames (list[Frame]): list of frames to be updated  
+            
+        Returns:
+            Data: The updated data.
+        """
+        if dataset_id is None:
+            raise BadParameterError("dataset_id is required.")
+        if data_id is None:
+            raise BadParameterError("data_id is required.")
+
+        response = self.request_gql(
+            Queries.UPDATE_FRAMES,
+            Queries.UPDATE_FRAMES["variables"](
+                dataset_id=dataset_id,
+                data_id=data_id,
+                frames=frames,
+            )
+        )
+        data = Data.model_validate(response)
+        return data

{superb_ai_onprem-0.4.5 → superb_ai_onprem-0.5.1}/spb_onprem/entities.py

@@ -6,6 +6,7 @@ from .data.entities import (
     Prediction,
     DataMeta,
     DataSlice,
+    Frame,
 )
 from .datasets.entities import Dataset
 from .slices.entities import Slice
@@ -41,6 +42,7 @@ __all__ = [
     "ActivityHistory",
     "Export",
     "Content",
+    "Frame",
 
     # Enums
     "DataType",

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.4.5
+Version: 0.5.1
 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.4.5 → superb_ai_onprem-0.5.1}/superb_ai_onprem.egg-info/SOURCES.txt

@@ -43,6 +43,7 @@ spb_onprem/data/entities/annotation.py
 spb_onprem/data/entities/data.py
 spb_onprem/data/entities/data_meta.py
 spb_onprem/data/entities/data_slice.py
+spb_onprem/data/entities/frame.py
 spb_onprem/data/entities/prediction.py
 spb_onprem/data/entities/scene.py
 spb_onprem/data/enums/__init__.py
@@ -72,6 +73,7 @@ spb_onprem/data/params/remove_data_meta.py
 spb_onprem/data/params/update_annotation.py
 spb_onprem/data/params/update_data.py
 spb_onprem/data/params/update_data_slice.py
+spb_onprem/data/params/update_frames.py
 spb_onprem/data/params/update_scene.py
 spb_onprem/data/params/update_slice_annotation.py
 spb_onprem/data/params/update_slice_annotation_version.py