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

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:
@@ -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.14.dist-info → rapidata-0.1.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.14
+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.14.dist-info → rapidata-0.1.15.dist-info}/RECORD

@@ -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=ucNWVUejksTLIboQYeqExnOuxmhEfhNj56Cuo59FYaU,7000
+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.14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-0.1.14.dist-info/METADATA,sha256=fJFwoFVeWplRXhdd4MWmc5HnLiCrojo_Htf-54CHDBo,768
-rapidata-0.1.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rapidata-0.1.14.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,,