rapidata 0.1.15__tar.gz → 0.1.17__tar.gz

{rapidata-0.1.15 → rapidata-0.1.17}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.1.15"
+version = "0.1.17"
 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.15 → rapidata-0.1.17}/rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -216,4 +216,8 @@ class RapidataOrderBuilder:
         :type amount: int
         :return: The updated RapidataOrderBuilder instance.
         """
-        self._selections.append(LabelingSelection(amount))
+        raise NotImplementedError("Not implemented yet.")
+
+    def selections(self, selections: list[Selection]):
+        self._selections = selections
+        return self

rapidata-0.1.17/rapidata/rapidata_client/selection/demographic_selection.py

@@ -0,0 +1,14 @@
+
+
+from rapidata.rapidata_client.selection.base_selection import Selection
+from rapidata.api_client.models.demographic_selection import DemographicSelection as DemographicSelectionModel
+
+
+class DemographicSelection(Selection):
+    """Demographic selection class."""
+
+    def __init__(self, rapid_id: str):
+        self.rapid_id = rapid_id
+
+    def to_model(self):
+        return DemographicSelectionModel(_t="DemographicSelection", rapidId=self.rapid_id)

rapidata-0.1.17/rapidata/rapidata_client/selection/validation_selection.py

@@ -0,0 +1,15 @@
+
+from rapidata.api_client.models.validation_selection import ValidationSelection as ValidationSelectionModel
+from rapidata.rapidata_client.selection.base_selection import Selection
+
+
+class ValidationSelection(Selection):
+
+    def __init__(self, validation_set_id: str, amount: int = 1):
+        self.validation_set_id = validation_set_id
+        self.amount = amount
+
+
+    def to_model(self):
+        return ValidationSelectionModel(_t="ValidationSelection", validationSetId=self.validation_set_id, amount=self.amount)
+