rapidata 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

rapidata/rapidata_client/dataset/rapidata_validation_set.py

@@ -14,6 +14,7 @@ from rapidata.api_client.models.bounding_box_truth import BoundingBoxTruth
 from rapidata.api_client.models.classify_payload import ClassifyPayload
 from rapidata.api_client.models.compare_payload import ComparePayload
 from rapidata.api_client.models.compare_truth import CompareTruth
+from rapidata.api_client.models.datapoint_metadata_model_metadata_inner import DatapointMetadataModelMetadataInner
 from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
 from rapidata.api_client.models.free_text_payload import FreeTextPayload
 from rapidata.api_client.models.line_payload import LinePayload
@@ -27,6 +28,7 @@ from rapidata.api_client.models.polygon_truth import PolygonTruth
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
 from rapidata.rapidata_client.dataset.validation_rapid_parts import ValidatioRapidParts
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.types import RapidAsset
 from rapidata.service.openapi_service import OpenAPIService
 
@@ -61,7 +63,7 @@ class RapidataValidationSet:
             | PolygonTruth
             | TranscriptionTruth
         ),
-        metadata: Any,
+        metadata: list[Metadata],
         media_paths: RapidAsset,
         randomCorrectProbability: float,
     ):
@@ -69,7 +71,7 @@ class RapidataValidationSet:
             validationSetId=self.id,
             payload=AddValidationRapidModelPayload(payload),
             truth=AddValidationRapidModelTruth(truths),
-            metadata=metadata or [],
+            metadata=[DatapointMetadataModelMetadataInner(meta.to_model()) for meta in metadata],
             randomCorrectProbability=randomCorrectProbability,
         )
 

rapidata/rapidata_client/dataset/validation_rapid_parts.py

@@ -19,6 +19,7 @@ from rapidata.api_client.models.polygon_payload import PolygonPayload
 from rapidata.api_client.models.polygon_truth import PolygonTruth
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.types import RapidAsset
 
 
@@ -48,5 +49,5 @@ class ValidatioRapidParts:
         | PolygonTruth
         | TranscriptionTruth
     )
-    metadata: Any
+    metadata: list[Metadata]
     randomCorrectProbability: float

rapidata/rapidata_client/dataset/validation_set_builder.py

@@ -1,13 +1,4 @@
-from dataclasses import dataclass
 import os
-from typing import Any, Union
-from rapidata.api_client.models.add_validation_rapid_model import AddValidationRapidModel
-from rapidata.api_client.models.add_validation_rapid_model_payload import (
-    AddValidationRapidModelPayload,
-)
-from rapidata.api_client.models.add_validation_rapid_model_truth import (
-    AddValidationRapidModelTruth,
-)
 from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
 from rapidata.api_client.models.classify_payload import ClassifyPayload
 from rapidata.api_client.models.compare_payload import ComparePayload
@@ -15,14 +6,14 @@ from rapidata.api_client.models.compare_truth import CompareTruth
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
 from rapidata.api_client.models.transcription_word import TranscriptionWord
-from rapidata.rapidata_client.dataset.rapidata_validation_set import RapidataValidationSet
+from rapidata.rapidata_client.dataset.rapidata_validation_set import (
+    RapidataValidationSet,
+)
 from rapidata.rapidata_client.dataset.validation_rapid_parts import ValidatioRapidParts
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.service.openapi_service import OpenAPIService
 
 
-
-
-
 class ValidationSetBuilder:
 
     def __init__(self, name: str, openapi_service: OpenAPIService):
@@ -42,7 +33,10 @@ class ValidationSetBuilder:
         if self.validation_set_id is None:
             raise ValueError("Failed to create validation set")
 
-        validation_set = RapidataValidationSet(validation_set_id=self.validation_set_id, openapi_service=self.openapi_service)
+        validation_set = RapidataValidationSet(
+            validation_set_id=self.validation_set_id,
+            openapi_service=self.openapi_service,
+        )
 
         for rapid_part in self._rapid_parts:
             validation_set.add_validation_rapid(
@@ -52,14 +46,19 @@ class ValidationSetBuilder:
                 media_paths=rapid_part.media_paths,
                 randomCorrectProbability=rapid_part.randomCorrectProbability,
             )
-            
+
         return validation_set
 
     def add_classify_rapid(
-        self, media_path: str, question: str, categories: list[str], truths: list[str]
+        self,
+        media_path: str,
+        question: str,
+        categories: list[str],
+        truths: list[str],
+        metadata: list[Metadata] = [],
     ):
         payload = ClassifyPayload(
-            _t="ClassifyPaylod", possibleCategories=categories, title=question
+            _t="ClassifyPayload", possibleCategories=categories, title=question
         )
         model_truth = AttachCategoryTruth(
             correctCategories=truths, _t="AttachCategoryTruth"
@@ -71,14 +70,20 @@ class ValidationSetBuilder:
                 media_paths=media_path,
                 payload=payload,
                 truths=model_truth,
-                metadata=None,
+                metadata=metadata,
                 randomCorrectProbability=len(truths) / len(categories),
             )
         )
 
         return self
 
-    def add_compare_rapid(self, media_paths: list[str], question: str, truth: str):
+    def add_compare_rapid(
+        self,
+        media_paths: list[str],
+        question: str,
+        truth: str,
+        metadata: list[Metadata] = [],
+    ):
         payload = ComparePayload(_t="ComparePayload", criteria=question)
         # take only last part of truth path
         truth = os.path.basename(truth)
@@ -91,7 +96,6 @@ class ValidationSetBuilder:
         for media_path in media_paths:
             if not os.path.exists(media_path):
                 raise FileNotFoundError(f"File not found: {media_path}")
-            
 
         self._rapid_parts.append(
             ValidatioRapidParts(
@@ -99,7 +103,7 @@ class ValidationSetBuilder:
                 media_paths=media_paths,
                 payload=payload,
                 truths=model_truth,
-                metadata=None,
+                metadata=metadata,
                 randomCorrectProbability=1 / len(media_paths),
             )
         )
@@ -113,6 +117,7 @@ class ValidationSetBuilder:
         transcription: list[str],
         correct_words: list[str],
         strict_grading: bool | None = None,
+        metadata: list[Metadata] = [],
     ):
         transcription_words = [
             TranscriptionWord(word=word, wordIndex=i)
@@ -143,7 +148,7 @@ class ValidationSetBuilder:
                 media_paths=media_path,
                 payload=payload,
                 truths=model_truth,
-                metadata=None,
+                metadata=metadata,
                 randomCorrectProbability=1 / len(transcription),
             )
         )

rapidata/rapidata_client/order/rapidata_order.py

@@ -1,10 +1,5 @@
 import time
-from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
-from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
 from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
-from rapidata.rapidata_client.workflow import Workflow
-from rapidata.api_client.models.create_order_model import CreateOrderModel
-from rapidata.rapidata_client.referee import Referee
 from rapidata.service.openapi_service import OpenAPIService
 
 
@@ -30,14 +25,6 @@ class RapidataOrder:
         self.order_id = order_id
         self._dataset = dataset
 
-    def create(self):
-        """
-        Creates the order using the provided name and workflow.
-
-        :return: The created RapidataOrder instance.
-        :rtype: RapidataOrder
-        """
-
     def submit(self):
         """
         Submits the order for processing.

rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -3,6 +3,9 @@ from rapidata.api_client.models.create_order_model import CreateOrderModel
 from rapidata.api_client.models.create_order_model_referee import (
     CreateOrderModelReferee,
 )
+from rapidata.api_client.models.create_order_model_selections_inner import (
+    CreateOrderModelSelectionsInner,
+)
 from rapidata.api_client.models.create_order_model_user_filters_inner import (
     CreateOrderModelUserFiltersInner,
 )
@@ -14,6 +17,8 @@ from rapidata.rapidata_client.feature_flags import FeatureFlags
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
 from rapidata.rapidata_client.referee.naive_referee import NaiveReferee
+from rapidata.rapidata_client.selection.base_selection import Selection
+from rapidata.rapidata_client.selection.labeling_selection import LabelingSelection
 from rapidata.rapidata_client.types import RapidAsset
 from rapidata.rapidata_client.workflow import Workflow
 from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
@@ -48,6 +53,7 @@ class RapidataOrderBuilder:
         self._validation_set_id: str | None = None
         self._feature_flags: FeatureFlags | None = None
         self._country_codes: list[str] | None = None
+        self._selections: list[Selection] = []
 
     def _to_model(self) -> CreateOrderModel:
         if self._workflow is None:
@@ -78,6 +84,10 @@ class RapidataOrderBuilder:
                 if self._feature_flags is not None
                 else None
             ),
+            selections=[
+                CreateOrderModelSelectionsInner(selection.to_model())
+                for selection in self._selections
+            ],
         )
 
     def create(self, submit=True) -> RapidataOrder:
@@ -186,7 +196,7 @@ class RapidataOrderBuilder:
         self._aggregator = aggregator
         return self
 
-    def validation_set(self, validation_set_id: str):
+    def validation_set_id(self, validation_set_id: str):
         """
         Set the validation set for the order.
 
@@ -197,3 +207,13 @@ class RapidataOrderBuilder:
         """
         self._validation_set_id = validation_set_id
         return self
+
+    def rapids_per_bag(self, amount: int):
+        """
+        Defines the number of tasks a user sees in a single session. The default is 3.
+
+        :param amount: The number of tasks a user sees in a single session.
+        :type amount: int
+        :return: The updated RapidataOrderBuilder instance.
+        """
+        self._selections.append(LabelingSelection(amount))

rapidata/rapidata_client/selection/base_selection.py

@@ -0,0 +1,9 @@
+from abc import abstractmethod
+from typing import Any
+
+
+class Selection:
+
+    @abstractmethod
+    def to_model(self) -> Any:
+        pass

rapidata/rapidata_client/selection/labeling_selection.py

@@ -0,0 +1,14 @@
+from typing import Any
+from rapidata.rapidata_client.selection.base_selection import Selection
+from rapidata.api_client.models.labeling_selection import (
+    LabelingSelection as LabelingSelectionModel,
+)
+
+
+class LabelingSelection(Selection):
+
+    def __init__(self, amount: int):
+        self.amount = amount
+
+    def to_model(self) -> Any:
+        return LabelingSelectionModel(_t="LabelingSelection", amount=self.amount)

{rapidata-0.1.13.dist-info → rapidata-0.1.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.13
+Version: 0.1.15
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG

{rapidata-0.1.13.dist-info → rapidata-0.1.15.dist-info}/RECORD

@@ -195,9 +195,9 @@ rapidata/rapidata_client/country_codes/__init__.py,sha256=Y8qeG2IMjvMGvhaPydq0nh
 rapidata/rapidata_client/country_codes/country_codes.py,sha256=Q0HMX7uHJQDeLCFPP5bq4iYi6pgcDWEcl2ONGhjgoeU,286
 rapidata/rapidata_client/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/dataset/rapidata_dataset.py,sha256=mVx-YU3N6lvr_0NCg7yGodeBJh8A_Y0tFWRrxcFg9-w,2506
-rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=nfaENQOJZW5mZBY-4WAt-DFb3o46nN0J-QnHCLvEFPA,3308
-rapidata/rapidata_client/dataset/validation_rapid_parts.py,sha256=96kb88XOwXuVI7stwY4Th4K3fo9-lFM-v7QV3qsS3NY,2073
-rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=UjKCgtQ23GM4CtiRrl1eQvD-eZFbiYCjdVf1iAVH8EI,5424
+rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=WvygSfVAFLd6Ha8uPgqWdS3pys5Kaf454hq0pGSsYao,3564
+rapidata/rapidata_client/dataset/validation_rapid_parts.py,sha256=UmH2yFXrN1uWTUDniMKJ03_mRC-sZBsvR4JhDUbKfJo,2153
+rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=C1mBZUr_ch_8_2mb5EfT44HFM4KaWIRSl1yLsGIsOus,5333
 rapidata/rapidata_client/feature_flags/__init__.py,sha256=BNG_NQ4CrrC61fAWliImr8r581pIvegrkepVVbxcBw8,55
 rapidata/rapidata_client/feature_flags/feature_flags.py,sha256=uV4TPXc1oyqBRC9Ch3APprZwxBGnGMAl9DMFBUWmvnA,1048
 rapidata/rapidata_client/metadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -206,13 +206,15 @@ rapidata/rapidata_client/metadata/private_text_metadata.py,sha256=VJ_SJwp8OMFZLu
 rapidata/rapidata_client/metadata/prompt_metadata.py,sha256=_FypjKWrC3iKUO_G2CVwAGcYbEYClVv4K7upqlnmecw,468
 rapidata/rapidata_client/metadata/transcription_metadata.py,sha256=THtDEVCON4UlcXHmXrjilaOLHys4TrktUOPGWnXaCcc,631
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rapidata/rapidata_client/order/rapidata_order.py,sha256=4a0Sn1y1UmL7HydhaDBOus-g_yILqPQMnTXfLEzQB5g,2997
-rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=JqPxaNxpeM4p5i-jghm9t34YLXwuy_a3abtwRh5553Y,6997
+rapidata/rapidata_client/order/rapidata_order.py,sha256=VRDLTPBf2k6UihF0DnWq3nfBLWExfWHzh3T-cJgFF1w,2437
+rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=iFIWwAo_fE887WNgykSHp6-2w8G9MIJZIjH07vGX8kQ,7856
 rapidata/rapidata_client/rapidata_client.py,sha256=BN3bfLEdNLYt7VZuEwDycNPT61wz66TTssjmNlRYlvU,2019
 rapidata/rapidata_client/referee/__init__.py,sha256=x0AxGCsR6TlDjfqQ00lB9V7QVS9EZCJzweNEIzx42PI,207
 rapidata/rapidata_client/referee/base_referee.py,sha256=bMy7cw0a-pGNbFu6u_1_Jplu0A483Ubj4oDQzh8vu8k,493
 rapidata/rapidata_client/referee/classify_early_stopping_referee.py,sha256=-mPTCck81r2xn1GtMoizaOCftA3cOfj6YDSFbqsI3Ic,1140
 rapidata/rapidata_client/referee/naive_referee.py,sha256=9CSoZou4YBFc9PiRvA2ELVQqyk-S1nug1SqhTwO9Tvw,719
+rapidata/rapidata_client/selection/base_selection.py,sha256=Y3HkROPm4I4HLNiR0HuHKpvk236KkRlsoDxQATm_chY,138
+rapidata/rapidata_client/selection/labeling_selection.py,sha256=cqDMQEXfQGMmgIvPgGOYgIGaXflV_J7LZsGOsakLXqo,425
 rapidata/rapidata_client/types/__init__.py,sha256=Q9lzemtloA8e_XVVV1Y7ZZGsNg-xRqgj1GdJPrCzo5A,33
 rapidata/rapidata_client/workflow/__init__.py,sha256=CDG8bKOBhLV4A0uBlqOd__79fH9KW3-UlZsR1RANLf0,250
 rapidata/rapidata_client/workflow/base_workflow.py,sha256=335UyiNsIFix7ru_KZLeQHvfNg3GSId44ppPmWTXs_Y,1267
@@ -225,7 +227,7 @@ rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5
 rapidata/service/openapi_service.py,sha256=xNlISI5NZByLy1Yeasc5Uafj1IA2mWJTzxUU1TtwyQU,1402
 rapidata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/utils/image_utils.py,sha256=TldO3eJWG8IhfJjm5MfNGO0mEDm1mQTsRoA0HLU1Uxs,404
-rapidata-0.1.13.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-0.1.13.dist-info/METADATA,sha256=rv6P4z8eZGGpSEFC80R3Xk6hInEpuO_au1IgU8GDfqs,768
-rapidata-0.1.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rapidata-0.1.13.dist-info/RECORD,,
+rapidata-0.1.15.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-0.1.15.dist-info/METADATA,sha256=y7chOvpw7mH-nVviDU3hZXWmxRC7Oh3kjLr_p5EVRwM,768
+rapidata-0.1.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rapidata-0.1.15.dist-info/RECORD,,