rapidata 0.1.14__tar.gz → 0.1.16__tar.gz

{rapidata-0.1.14 → rapidata-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.14
+Version: 0.1.16
 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 → rapidata-0.1.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.1.14"
+version = "0.1.16"
 description = "Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way."
 authors = ["Rapidata AG <info@rapidata.ai>"]
 license = "Apache-2.0"

{rapidata-0.1.14 → rapidata-0.1.16}/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-0.1.14 → rapidata-0.1.16}/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,14 @@ 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))
+        return self

rapidata-0.1.16/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-0.1.16/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)