superb-ai-onprem 0.8.3__py3-none-any.whl → 0.9.1__py3-none-any.whl

spb_onprem/__init__.py

@@ -10,7 +10,6 @@ from .slices.service import SliceService
 from .activities.service import ActivityService
 from .exports.service import ExportService
 from .contents.service import ContentService
-from .predictions.service import PredictionService
 from .models.service import ModelService
 from .inferences.service import InferService
 
@@ -21,7 +20,6 @@ from .entities import (
     Scene,
     Annotation,
     AnnotationVersion,
-    Prediction,
     DataMeta,
     Dataset,
     Slice,
@@ -90,7 +88,6 @@ __all__ = (
     "ActivityService",
     "ExportService",
     "ContentService",
-    "PredictionService",
     "ModelService",
     "InferService",
 
@@ -99,7 +96,6 @@ __all__ = (
     "Scene",
     "Annotation",
     "AnnotationVersion",
-    "Prediction",
     "DataMeta",
     "Dataset",
     "Slice",
@@ -109,7 +105,6 @@ __all__ = (
     "Export",
     "Content",
     "Frame",
-    "PredictionSet",
     "Model",
     "Comment",
     "Reply",

spb_onprem/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.8.3'
-__version_tuple__ = version_tuple = (0, 8, 3)
+__version__ = version = '0.9.1'
+__version_tuple__ = version_tuple = (0, 9, 1)
 
 __commit_id__ = commit_id = None

spb_onprem/data/entities/__init__.py

@@ -2,7 +2,6 @@ from .annotation import Annotation, AnnotationVersion
 from .comment import Comment, Reply
 from .data_meta import DataMeta
 from .data import Data
-from .prediction import Prediction
 from .scene import Scene
 from .data_slice import DataSlice
 from .frame import Frame
@@ -16,7 +15,6 @@ __all__ = (
     "AnnotationVersion",
     "Comment",
     "Reply",
-    "Prediction",
     "DataMeta",
     "DataSlice",
 )

spb_onprem/data/entities/data.py

@@ -3,7 +3,6 @@ from spb_onprem.base_model import CustomBaseModel, Field
 from spb_onprem.data.enums import DataType
 from .scene import Scene
 from .annotation import Annotation
-from .prediction import Prediction
 from .data_meta import DataMeta
 from .data_slice import DataSlice
 from .frame import Frame
@@ -19,7 +18,6 @@ class Data(CustomBaseModel):
     # 식별자
     id: Optional[str] = Field(None, description="데이터 고유 식별자")
     dataset_id: Optional[str] = Field(None, alias="datasetId", description="상위 데이터셋 ID")
-    slice_ids: Optional[List[str]] = Field(None, alias="sliceIds", description="소속된 슬라이스 ID 목록")
     key: Optional[str] = Field(None, description="사용자 정의 고유 키")
     
     # 데이터 타입 및 내용
@@ -30,8 +28,7 @@ class Data(CustomBaseModel):
     
     # 어노테이션 및 예측
     annotation: Optional[Annotation] = Field(None, description="어노테이션 데이터")
-    predictions: Optional[List[Prediction]] = Field(None, description="ML 모델 예측 결과")
-    
+
     # 메타데이터
     meta: Optional[List[DataMeta]] = Field(None, description="커스텀 메타데이터 목록")
     

spb_onprem/data/params/__init__.py

@@ -37,9 +37,6 @@ from .data_list import (
 from .get_data_detail import (
     get_data_detail_params,
 )
-from .get_evaluation_value_list import (
-    get_evaluation_value_list_params,
-)
 from .remove_data_from_slice import (
     remove_data_from_slice_params
 )
@@ -49,12 +46,6 @@ from .insert_data_to_slice import (
 from .delete_data import (
     delete_data_params
 )
-from .insert_prediction import (
-    insert_prediction_params
-)
-from .delete_prediction import (
-    delete_prediction_params
-)
 from .update_annotation import (
     update_annotation_params
 )
@@ -99,7 +90,6 @@ __all__ = [
     "get_data_id_list_params",
     "get_data_list_params",
     "get_data_detail_params",
-    "get_evaluation_value_list_params",
     "DateTimeRangeFilterOption",
     "UserFilterOption",
     "NumericRangeFilter",
@@ -126,8 +116,6 @@ __all__ = [
     "remove_data_from_slice_params",
     "insert_data_to_slice_params",
     "delete_data_params",
-    "insert_prediction_params",
-    "delete_prediction_params",
     "update_annotation_params",
     "insert_annotation_version_params",
     "delete_annotation_version_params",

spb_onprem/data/params/create_data.py

@@ -13,7 +13,7 @@ def create_params(
         "datasetId": data.dataset_id,
         "key": data.key,
         "type": data.type.value,
-        "slices": data.slice_ids if data.slice_ids is not None else [],
+        "slices": [slice.id for slice in data.slices] if data.slices is not None else [],
         "scene": [
             {
                 "type": scene.type.value,
@@ -39,16 +39,6 @@ def create_params(
             ] if data.annotation.versions is not None else [],
             "meta": data.annotation.meta
         } if data.annotation is not None else None,
-        "predictions": [
-            {
-                "setId": prediction.set_id,
-                "content": {
-                    "id": prediction.content.id,
-                },
-                "meta": prediction.meta,
-            }
-            for prediction in data.predictions
-        ] if data.predictions is not None else [],
         "meta": [
             {
                 "key": meta.key,

spb_onprem/data/queries.py

@@ -5,12 +5,9 @@ from .params import (
     get_data_id_list_params,
     get_data_list_params,
     get_data_detail_params,
-    get_evaluation_value_list_params,
     remove_data_from_slice_params,
     insert_data_to_slice_params,
     delete_data_params,
-    insert_prediction_params,
-    delete_prediction_params,
     update_annotation_params,
     insert_annotation_version_params,
     delete_annotation_version_params,
@@ -42,7 +39,6 @@ class Schemas:
     DATA = '''
         id
         datasetId
-        sliceIds
         key
         type
         scene {
@@ -120,13 +116,6 @@ class Schemas:
             }
             meta
         }
-        predictions {
-            setId
-            content {
-                id
-            }
-            meta
-        }
         thumbnail {
             id
         }
@@ -155,7 +144,6 @@ class Queries():
                 $scene: [SceneInput!],
                 $thumbnail: ContentBaseInput,
                 $annotation: AnnotationInput,
-                $predictions: [PredictionInput!],
                 $meta: [DataMetaInput!]
             ) {{
             createData(
@@ -166,7 +154,6 @@ class Queries():
                 scene: $scene,
                 thumbnail: $thumbnail,
                 annotation: $annotation,
-                predictions: $predictions,
                 meta: $meta,
             ) 
                 {{
@@ -325,46 +312,6 @@ class Queries():
         ''',
         "variables": delete_data_params,
     }
-
-    INSERT_PREDICTION = {
-        "name": "insertPrediction",
-        "query": f'''
-            mutation (
-                $dataset_id: ID!,
-                $data_id: ID!,
-                $prediction: PredictionInput!,
-            ) {{
-                insertPrediction(
-                    datasetId: $dataset_id,
-                    dataId: $data_id,
-                    prediction: $prediction,
-                ) {{
-                    {Schemas.DATA}
-                }}
-            }}
-        ''',
-        "variables": insert_prediction_params,
-    }
-
-    DELETE_PREDICTION = {
-        "name": "deletePrediction",
-        "query": f'''
-            mutation (
-                $dataset_id: ID!,
-                $data_id: ID!,
-                $set_id: ID!,
-            ) {{
-                deletePrediction(
-                    datasetId: $dataset_id,
-                    dataId: $data_id,
-                    setId: $set_id,
-                ) {{
-                    {Schemas.DATA}
-                }}
-            }}
-        ''',
-        "variables": delete_prediction_params,
-    }
     
     UPDATE_ANNOTATION = {
         "name": "updateAnnotation",
@@ -627,31 +574,3 @@ class Queries():
         ''',
         "variables": update_frames_params,
     }
-    
-    GET_EVALUATION_VALUE_LIST = {
-        "name": "evaluationValueList",
-        "query": '''
-            query GetEvaluationValueList(
-                $datasetId: String!,
-                $predictionSetId: String!,
-                $filter: DiagnosisFilter,
-                $length: Int,
-                $cursor: String
-            ) {
-                evaluationValueList(
-                    datasetId: $datasetId,
-                    predictionSetId: $predictionSetId,
-                    filter: $filter,
-                    length: $length,
-                    cursor: $cursor
-                ) {
-                    totalCount
-                    next
-                    data {
-                        dataId
-                    }
-                }
-            }
-        ''',
-        "variables": get_evaluation_value_list_params
-    }

spb_onprem/data/service.py

@@ -17,7 +17,6 @@ from .queries import Queries
 from .entities import (
     Data,
     AnnotationVersion,
-    Prediction,
     Frame,
     DataMeta,
 )
@@ -297,60 +296,6 @@ class DataService(BaseService):
         )
         return response
 
-    def insert_prediction(
-        self,
-        dataset_id: str,
-        data_id: str,
-        prediction: Prediction,
-    ):
-        """Insert a prediction.
-
-        Args:
-            dataset_id (str): The dataset id.
-            data_id (str): The data id.
-            prediction (Prediction): The prediction.
-
-        Returns:
-            Data: The updated data.
-        """
-        response = self.request_gql(
-            Queries.INSERT_PREDICTION,
-            Queries.INSERT_PREDICTION["variables"](
-                dataset_id=dataset_id,
-                data_id=data_id,
-                prediction=prediction,
-            )
-        )
-        data = Data.model_validate(response)
-        return data
-    
-    def delete_prediction(
-        self,
-        dataset_id: str,
-        data_id: str,
-        set_id: str,
-    ):
-        """Delete a prediction.
-
-        Args:
-            dataset_id (str): The dataset id.
-            data_id (str): The data id.
-            set_id (str): The set id.
-
-        Returns:
-            Data: The updated data.
-        """
-        response = self.request_gql(
-            Queries.DELETE_PREDICTION,
-            Queries.DELETE_PREDICTION["variables"](
-                dataset_id=dataset_id,
-                data_id=data_id,
-                set_id=set_id,
-            )
-        )
-        data = Data.model_validate(response)
-        return data
-    
     def update_annotation(
         self,
         dataset_id: str,
@@ -781,66 +726,3 @@ class DataService(BaseService):
         )
         data = Data.model_validate(response)
         return data
-    
-    def get_evaluation_value_list(
-        self,
-        dataset_id: str,
-        prediction_set_id: str,
-        filter: Union[UndefinedType, dict] = Undefined,
-        length: int = 50,
-        cursor: Union[UndefinedType, str] = Undefined
-    ) -> dict:
-        """Get evaluation values list for diagnosis filtering.
-        
-        Retrieves evaluation values for diagnosis filtering with pagination support.
-        
-        Args:
-            dataset_id (str): The dataset ID.
-            prediction_set_id (str): The prediction set ID.
-            filter (Union[UndefinedType, dict]): Diagnosis filter for evaluation values.
-            length (int): Number of items to retrieve per page.
-            cursor (Union[UndefinedType, str]): Cursor for pagination.
-        
-        Returns:
-            dict: Response containing totalCount, next cursor, and data list with dataId fields.
-        """
-        if dataset_id is None:
-            raise BadParameterError("dataset_id is required.")
-        if prediction_set_id is None:
-            raise BadParameterError("prediction_set_id is required.")
-
-        response = self.request_gql(
-            Queries.GET_EVALUATION_VALUE_LIST,
-            Queries.GET_EVALUATION_VALUE_LIST["variables"](
-                dataset_id=dataset_id,
-                prediction_set_id=prediction_set_id,
-                filter=filter,
-                length=length,
-                cursor=cursor
-            )
-        )
-        return response.get("evaluationValueList", {})
-    
-    def get_total_data_id_count_in_evaluation_value(
-        self,
-        dataset_id: str,
-        prediction_set_id: str,
-        filter: Union[UndefinedType, dict] = Undefined
-    ) -> int:
-        """Get total count of data IDs in evaluation values for diagnosis filtering.
-        
-        Args:
-            dataset_id (str): The dataset ID.
-            prediction_set_id (str): The prediction set ID.
-            filter (Union[UndefinedType, dict]): Diagnosis filter for evaluation values.
-        
-        Returns:
-            int: Total count of evaluation values.
-        """
-        result = self.get_evaluation_value_list(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id,
-            filter=filter,
-            length=1
-        )
-        return result.get("totalCount", 0)

spb_onprem/entities.py

@@ -3,7 +3,6 @@ from .data.entities import (
     Scene,
     Annotation,
     AnnotationVersion,
-    Prediction,
     DataMeta,
     DataSlice,
     Frame,
@@ -29,7 +28,6 @@ from .activities.entities import (
 )
 from .exports.entities import Export
 from .contents.entities import Content
-from .predictions.entities import PredictionSet
 from .models.entities import Model
 
 __all__ = [
@@ -38,7 +36,6 @@ __all__ = [
     "Scene",
     "Annotation",
     "AnnotationVersion",
-    "Prediction",
     "DataMeta",
     "DataSlice",
     "Dataset",
@@ -48,7 +45,6 @@ __all__ = [
     "Export",
     "Content",
     "Frame",
-    "PredictionSet",
     "Model",
     "Comment",
     "Reply",

{superb_ai_onprem-0.8.3.dist-info → superb_ai_onprem-0.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.8.3
+Version: 0.9.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.8.3.dist-info → superb_ai_onprem-0.9.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-spb_onprem/__init__.py,sha256=Q5aooIhNiuf0z9-S5k5FBfM_oA_c92kCzWdjLuDX4wE,3294
-spb_onprem/_version.py,sha256=vG9GelPWtHXRUy7hmX4yak1DOSnho8XScbsq70cQTPY,704
+spb_onprem/__init__.py,sha256=JZHiLuMnbQA-xFapkImbFEA8quPbG2u_iyySTMq1ENQ,3163
+spb_onprem/_version.py,sha256=LwGndsRSpclYq-j3wgRr2nzOXwUYj0Jtg7Kof7R0BEw,704
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=_6aLM_zn4Ycd7nAc_BA395KmxH5lcSxd7gmgG9DdOf4,6518
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=rpkJORjr5d6Mwmt-PU04z_6JOj7w_bYqZyf9zEKcOnA,1206
+spb_onprem/entities.py,sha256=bgRWSF_ksfAD6LxVpvEx5eAHqEnFIZL4Xh2Lw2uxvi4,1103
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
 spb_onprem/searches.py,sha256=D6FUAf0cpjkRxNOBnz3HDFcnVFxSu2B6xkHMWKF0dOU,1754
 spb_onprem/activities/__init__.py,sha256=iTHUza8mtiBd_QEPH-0tzrOsd_VSwSQNMbzVWfebQ6c,175
@@ -33,12 +33,12 @@ spb_onprem/contents/params/delete_content.py,sha256=_M9d6FJpA47_pcM5YyHbMmTZv1F2
 spb_onprem/contents/params/get_download_url.py,sha256=3oaiqmjJRxw0U59SoTxoih1pcU-cuQvcU-g89lCcqtQ,98
 spb_onprem/contents/params/get_upload_url.py,sha256=j6s39Y796yJBTtgAbWrW7Rr1PK__XHViMq3JLEaL9xQ,263
 spb_onprem/data/__init__.py,sha256=5XKxNm2BlKufrX5uRQreUEzJ-nerTrozKpG1RJL5wt8,68
-spb_onprem/data/queries.py,sha256=0R8XosMWZPidDICmhGsSpyQNGnuMVFtSHQVH7AZPqXg,17144
-spb_onprem/data/service.py,sha256=I6DkcltPe08OBxpBQ-9rOIWzDlIqPn59et1J4PhHm0c,25098
-spb_onprem/data/entities/__init__.py,sha256=3TWoFmEzbGTxRKgxzDFzQLIygGWXsGf1NcPiT7fhqXw,440
+spb_onprem/data/queries.py,sha256=2nlE18kbQsdM9MlgSlSLkh2d9x6bnB3LkTTGhhAYxps,14885
+spb_onprem/data/service.py,sha256=R_cfMPQJFYI9x_6nACv3Et8M5QceLyYB5XVX6fJeOkE,21351
+spb_onprem/data/entities/__init__.py,sha256=7rrLybW_FPj_tJDYbtj88VMARSMCcCPywV8D0LHFm1U,387
 spb_onprem/data/entities/annotation.py,sha256=o6ctyXWXDPTmI97W7B1RKcqDNlZIZjrlGg-MgHavB6w,1432
 spb_onprem/data/entities/comment.py,sha256=l7o232sfuJifbES76KPsD7Su4I58ORlZ49Wum_uNxeg,2050
-spb_onprem/data/entities/data.py,sha256=tUyKVrPgycD6-wQonGBXqse3AUux-VzlgePBFBz9DwE,2367
+spb_onprem/data/entities/data.py,sha256=hcLhbUFMBSwoXIHisLcgeQPqg6X3iey52RdMWhA7uv8,2116
 spb_onprem/data/entities/data_meta.py,sha256=Cm9JQ4nv8LZwJaKcmNxxvuRUOvncKnzHW0cpoLYUAMA,1044
 spb_onprem/data/entities/data_slice.py,sha256=Fyr7iQRs2iET4Hp0DIUNQD1Qed3q46408gc9-GtDWbI,1345
 spb_onprem/data/entities/frame.py,sha256=ue62YvdKg0L-561zgOB1KrMi5H3x-I99larcFk7CW9s,990
@@ -49,23 +49,20 @@ spb_onprem/data/enums/data_meta_type.py,sha256=vFPAZACPtFrJnzCW_F6ZheWa-Tc5w0yNv
 spb_onprem/data/enums/data_status.py,sha256=uV_yLtigSmht-tKFaLwvO4MpV5j4GzwuBRDILGrxMBA,292
 spb_onprem/data/enums/data_type.py,sha256=9ErNmOK-RIOrhyt25SE2WVPSx_nTxt3ncTG5UtW92iM,225
 spb_onprem/data/enums/scene_type.py,sha256=pTibjST4nJc3ylUrjQifYBdtLIcS2D0q71bdhyDjbIk,205
-spb_onprem/data/params/__init__.py,sha256=Haiq-OCMEngWSxXmYKuTpp4DApN58ffGltFp61s83eI,3546
+spb_onprem/data/params/__init__.py,sha256=Rp63SIbYikYBwJ_K0RU7XGoEuOezVberRLehXZsF9HM,3233
 spb_onprem/data/params/change_data_labeler.py,sha256=jUG6bm0uheR1jVCy-RwFY97C9K93fdDdf3KDHJIelM4,570
 spb_onprem/data/params/change_data_reviewer.py,sha256=dFix3NfS2_8lpaoWxz6anLEJWMLdmRKM34ib8XwQiVQ,577
 spb_onprem/data/params/change_data_status.py,sha256=34gGtmTJT8wiDDpHOgQYvgP0-ziuYEBDFlpfY8SAZtI,593
-spb_onprem/data/params/create_data.py,sha256=aL6Bwd9talB2rnmdHbR9QkUx1Msni07-6aoBkhGoivM,1827
+spb_onprem/data/params/create_data.py,sha256=T3WhIfQUR0TZxY9o5dsM9yIJ6K7_gQ74fPjZv9f7Rao,1512
 spb_onprem/data/params/data.py,sha256=ZPJRjCbkjOCV73NxlGm5DxbSaNOAztVBRZRE8FFQ9mw,779
 spb_onprem/data/params/data_list.py,sha256=1pMIIYweBWx55Yvf2JWORfmBjzMCbGK_YYQEvT4CSSs,15193
 spb_onprem/data/params/delete_annotation_version.py,sha256=R_jBVujor-09VS1Aa8yGP14R8nJ2Aa9OmmKezC4gz4c,457
 spb_onprem/data/params/delete_data.py,sha256=D0tXTVbQogAhB49dHIxrX1r934kbtaJ_HOylJhLZVzA,324
-spb_onprem/data/params/delete_prediction.py,sha256=X3sV2mrSUqg8FOlZYRtb0o1hAwFos48ydFK2Il8UCp4,479
 spb_onprem/data/params/delete_scene.py,sha256=pD8TtS5Jm2uQC3F2EQ6-nAbQuQbJLG2pAwFqmsgucyM,478
 spb_onprem/data/params/delete_slice_annotation_version.py,sha256=5M12bHKg9HJlh6hQ8XgLwFrfhTU7CiLM__gLZVBYrMg,595
 spb_onprem/data/params/get_data_detail.py,sha256=7_aWRcHHsX0U2chl3CWrVLRUfzeoPzK8JtjwUGQdv0Y,388
-spb_onprem/data/params/get_evaluation_value_list.py,sha256=zsnk5y2FrZXHbb8jIsxQ8BTTX9-HIAVXZe4B37mxvmk,1115
 spb_onprem/data/params/insert_annotation_version.py,sha256=bxw9LndaDJ0DLBo8U8kkGctHz1ToEgD_UxWWVpmJpks,987
 spb_onprem/data/params/insert_data_to_slice.py,sha256=_hLNC80wJbxGSouOpjwTY19mAf-itfRR4sxHnlLiiqk,478
-spb_onprem/data/params/insert_prediction.py,sha256=68q5ZSSq4arAsozy1i52phkUoBfUSzNVVxCuygRYDfI,621
 spb_onprem/data/params/insert_scene.py,sha256=n36XVbOlPc8MrFusOtXaXDHmG2Wa13ZnQX-0300b12E,774
 spb_onprem/data/params/insert_slice_annotation_version.py,sha256=IRiwWbzIAcHWRiBNU-5F7qFCm4Cq78i2boWFSiAsR58,1052
 spb_onprem/data/params/remove_data_from_slice.py,sha256=UjoQH0gTdm1nE1DafUmq3CP1nzHiCwDUnytQ6oBcZAA,478
@@ -134,7 +131,7 @@ spb_onprem/slices/params/update_slice.py,sha256=kryOmCnRTQ_OU0qDKgugppLrpeUpuLwm
 spb_onprem/users/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spb_onprem/users/entities/__init__.py,sha256=X8HZsCTlQnuPszok3AwI-i7bsQi0Ehul5L_2jZaol5E,57
 spb_onprem/users/entities/auth.py,sha256=_KP-7yUErBxhJMm-dE3ObprPEG6e0JI2qNg6g8aK1qM,3371
-superb_ai_onprem-0.8.3.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.9.1.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
 tests/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/real_test.py,sha256=0gQHg7rIEuZndGZyNHMWSD5nUZPMsUGigfCjWFxMthQ,1786
@@ -142,7 +139,7 @@ tests/activities/test_get_activity_history.py,sha256=gNLc4OQLgw9wqToRxV0y5REAHPu
 tests/activities/test_params.py,sha256=P_k03Ll-We3gyFTeACalErP1UzwS2gB6ayIerKMMLEU,2335
 tests/activities/test_service.py,sha256=8nijwzJlAFL9S0TqyuAvOJZzBvPPjxJbrj3EyziT_Ng,4407
 tests/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/data/test_data_service.py,sha256=Lgh74J6lI3PhBNwAIbvd5brgmT0ktEOf3MclGIYjAEE,16823
+tests/data/test_data_service.py,sha256=DFIoI0ekAIDfv2k4FBGetOdMBZz68B6WyVmQygkMCec,9985
 tests/datasets/__init__.py,sha256=nMsjhPXe9rU87gI_wasqCtz7SsTae2LzpUUnXXzS5zI,23
 tests/datasets/test_dataset_service.py,sha256=RYY60U2TwKNv5JWzzkgUfP8ZgT5UYvgTYOl_fyp-Gek,4807
 tests/exports/__init__.py,sha256=iBqr781TtnT1Tm5wLTCsdftpuRgQifIibxWrB_I7tiw,23
@@ -155,7 +152,7 @@ tests/models/__init__.py,sha256=Z2naiUgIjI-nHGsldwze1BqpV5JfDKTaBk-vg6_h6t4,21
 tests/models/test_model_service.py,sha256=lQKFPHlfxhqcnrKHz6l2sGp53lpmC_kCHgQ-sXBbgMw,7556
 tests/predictions/__init__.py,sha256=mFAXy0xOEXVYM7quYfIhwSih3kcEDdD4rLPq5G91Igg,26
 tests/predictions/test_prediction_service.py,sha256=u2MYXNXSXKRAbv-Rz7sV0roitbD2h5U5ZIAaqeFMA1M,12632
-superb_ai_onprem-0.8.3.dist-info/METADATA,sha256=-eEcFrqXkRK6MVEQE6SYDyYzU7SX4QVRZwXbWPNbmis,9627
-superb_ai_onprem-0.8.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.8.3.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.8.3.dist-info/RECORD,,
+superb_ai_onprem-0.9.1.dist-info/METADATA,sha256=XnYNxEQdA1w0MoijwLSEvUbyqliFEXTpcKaFsyiFNhU,9627
+superb_ai_onprem-0.9.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.9.1.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.9.1.dist-info/RECORD,,

tests/data/test_data_service.py

@@ -38,12 +38,6 @@ class TestDataService:
                         }
                     ]
                 },
-                "predictions": [
-                    {
-                        "id": "prediction-1",
-                        "content": {"id": "content-3"}
-                    }
-                ],
                 "thumbnail": {
                     "id": "thumbnail-1"
                 }
@@ -176,12 +170,6 @@ class TestDataService:
                         }
                     ]
                 },
-                "predictions": [
-                    {
-                        "id": "prediction-1",
-                        "content": {"id": "prediction-content-1"}
-                    }
-                ],
                 "thumbnail": {
                     "id": "thumbnail-complex"
                 }
@@ -197,7 +185,6 @@ class TestDataService:
         assert result.id == data_id
         assert len(result.scene) == 2
         assert len(result.annotation.versions) == 2
-        assert len(result.predictions) == 1
         assert result.thumbnail.id == "thumbnail-complex"
 
     def test_get_data_detail_exception_handling(self):
@@ -213,204 +200,6 @@ class TestDataService:
 
         self.data_service.request_gql.assert_called_once()
 
-    def test_get_evaluation_value_list_success(self):
-        """Test successful evaluation value list retrieval."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        filter_dict = {"score": {"min": 0.8}}
-        length = 25
-        cursor = "cursor-abc"
-        
-        mock_response = {
-            "evaluationValueList": {
-                "totalCount": 100,
-                "next": "cursor-def",
-                "data": [
-                    {"dataId": "data-1"},
-                    {"dataId": "data-2"},
-                    {"dataId": "data-3"}
-                ]
-            }
-        }
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_evaluation_value_list(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id,
-            filter=filter_dict,
-            length=length,
-            cursor=cursor
-        )
-
-        # Assert
-        expected_result = {
-            "totalCount": 100,
-            "next": "cursor-def",
-            "data": [
-                {"dataId": "data-1"},
-                {"dataId": "data-2"},
-                {"dataId": "data-3"}
-            ]
-        }
-        assert result == expected_result
-        self.data_service.request_gql.assert_called_once_with(
-            Queries.GET_EVALUATION_VALUE_LIST,
-            Queries.GET_EVALUATION_VALUE_LIST["variables"](
-                dataset_id=dataset_id,
-                prediction_set_id=prediction_set_id,
-                filter=filter_dict,
-                length=length,
-                cursor=cursor
-            )
-        )
-
-    def test_get_evaluation_value_list_without_filter(self):
-        """Test evaluation value list retrieval without filter."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        
-        mock_response = {
-            "evaluationValueList": {
-                "totalCount": 50,
-                "next": None,
-                "data": [{"dataId": "data-1"}]
-            }
-        }
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_evaluation_value_list(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id
-        )
-
-        # Assert
-        assert result["totalCount"] == 50
-        assert result["next"] is None
-        assert len(result["data"]) == 1
-
-    def test_get_evaluation_value_list_missing_dataset_id(self):
-        """Test evaluation value list with missing dataset_id."""
-        # Arrange
-        prediction_set_id = "pred-set-456"
-
-        # Act & Assert
-        with pytest.raises(BadParameterError, match="dataset_id is required"):
-            self.data_service.get_evaluation_value_list(
-                dataset_id=None,
-                prediction_set_id=prediction_set_id
-            )
-
-    def test_get_evaluation_value_list_missing_prediction_set_id(self):
-        """Test evaluation value list with missing prediction_set_id."""
-        # Arrange
-        dataset_id = "dataset-123"
-
-        # Act & Assert
-        with pytest.raises(BadParameterError, match="prediction_set_id is required"):
-            self.data_service.get_evaluation_value_list(
-                dataset_id=dataset_id,
-                prediction_set_id=None
-            )
-
-    def test_get_evaluation_value_list_empty_response(self):
-        """Test evaluation value list with empty response."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        
-        mock_response = {}
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_evaluation_value_list(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id
-        )
-
-        # Assert
-        assert result == {}
-
-    def test_get_total_data_id_count_in_evaluation_value_success(self):
-        """Test successful total count retrieval."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        filter_dict = {"score": {"min": 0.8}}
-        
-        mock_response = {
-            "evaluationValueList": {
-                "totalCount": 150,
-                "next": "cursor-abc",
-                "data": [{"dataId": "data-1"}]
-            }
-        }
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_total_data_id_count_in_evaluation_value(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id,
-            filter=filter_dict
-        )
-
-        # Assert
-        assert result == 150
-        self.data_service.request_gql.assert_called_once_with(
-            Queries.GET_EVALUATION_VALUE_LIST,
-            Queries.GET_EVALUATION_VALUE_LIST["variables"](
-                dataset_id=dataset_id,
-                prediction_set_id=prediction_set_id,
-                filter=filter_dict,
-                length=1
-            )
-        )
-
-    def test_get_total_data_id_count_in_evaluation_value_zero_count(self):
-        """Test total count retrieval with zero results."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        
-        mock_response = {
-            "evaluationValueList": {
-                "totalCount": 0,
-                "next": None,
-                "data": []
-            }
-        }
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_total_data_id_count_in_evaluation_value(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id
-        )
-
-        # Assert
-        assert result == 0
-
-    def test_get_total_data_id_count_in_evaluation_value_missing_response(self):
-        """Test total count retrieval with missing response fields."""
-        # Arrange
-        dataset_id = "dataset-123"
-        prediction_set_id = "pred-set-456"
-        
-        mock_response = {}
-        self.data_service.request_gql.return_value = mock_response
-
-        # Act
-        result = self.data_service.get_total_data_id_count_in_evaluation_value(
-            dataset_id=dataset_id,
-            prediction_set_id=prediction_set_id
-        )
-
-        # Assert
-        assert result == 0
-
     def test_delete_data_query_structure(self):
         """Test DELETE query structure and parameters."""
         # Arrange

spb_onprem/data/params/delete_prediction.py

@@ -1,22 +0,0 @@
-
-
-def delete_prediction_params (
-    dataset_id: str,
-    data_id: str,
-    set_id: str,
-):
-    """Delete prediction from selected data.
-
-    Args:
-        dataset_id (str): dataset id which the data belongs to
-        data_id (str): data id to be deleted
-        set_id (str): set id to be deleted
-        
-    Returns:
-        dict: the params for graphql query
-    """
-    return {
-        "dataset_id": dataset_id,
-        "data_id": data_id,
-        "set_id": set_id,
-    }

spb_onprem/data/params/get_evaluation_value_list.py

@@ -1,36 +0,0 @@
-from typing import Union
-from spb_onprem.base_types import UndefinedType, Undefined
-
-
-def get_evaluation_value_list_params(
-    dataset_id: str,
-    prediction_set_id: str,
-    filter: Union[UndefinedType, dict] = Undefined,
-    length: int = 50,
-    cursor: Union[UndefinedType, str] = Undefined
-):
-    """Generate variables for get evaluation value list GraphQL query.
-    
-    Args:
-        dataset_id (str): The ID of the dataset.
-        prediction_set_id (str): The ID of the prediction set.
-        filter (Union[UndefinedType, dict], optional): Diagnosis filter for evaluation values.
-        length (int): Number of items to retrieve per page.
-        cursor (Union[UndefinedType, str], optional): Cursor for pagination.
-        
-    Returns:
-        dict: Variables dictionary for the GraphQL query.
-    """
-    params = {
-        "datasetId": dataset_id,
-        "predictionSetId": prediction_set_id,
-        "length": length
-    }
-    
-    if filter is not Undefined:
-        params["filter"] = filter
-        
-    if cursor is not Undefined:
-        params["cursor"] = cursor
-        
-    return params

spb_onprem/data/params/insert_prediction.py

@@ -1,25 +0,0 @@
-from spb_onprem.data.entities import Prediction
-
-
-def insert_prediction_params (
-    dataset_id: str,
-    data_id: str,
-    prediction: Prediction,
-):
-    """Insert prediction to selected data.
-
-    Args:
-        dataset_id (str): dataset id which the data belongs to
-        data_id (str): data id to be inserted
-        prediction (Prediction): prediction to be inserted
-
-    Returns:
-        dict: the params for graphql query
-    """
-    return {
-        "dataset_id": dataset_id,
-        "data_id": data_id,
-        "prediction": prediction.model_dump(
-            by_alias=True, exclude_unset=True
-        ),
-    }