rapidata 0.1.8__tar.gz → 0.1.10__tar.gz

{rapidata-0.1.8 → rapidata-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.8
+Version: 0.1.10
 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
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: PyJWT (>=2.9.0,<3.0.0)
 Requires-Dist: pillow (>=10.4.0,<11.0.0)
+Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Description-Content-Type: text/markdown
 

{rapidata-0.1.8 → rapidata-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.1.8"
+version = "0.1.10"
 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"
@@ -11,6 +11,7 @@ python = "^3.12"
 requests = "^2.32.3"
 PyJWT = "^2.9.0"
 pillow = "^10.4.0"
+pydantic = "^2.8.2"
 
 [tool.poetry.group.dev.dependencies]
 python-dotenv = "^1.0.1"

rapidata-0.1.10/rapidata/rapidata_client/feature_flags/feature_flags.py

@@ -0,0 +1,33 @@
+from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
+
+
+class FeatureFlags:
+    def __init__(self):
+        self._flags: dict[str, str] = {}
+
+    def to_list(self) -> list[FeatureFlagModel]:
+        return [FeatureFlagModel(key=name, value=value) for name, value in self._flags.items()]
+
+    def alert_on_fast_response(self, value: int):
+        self._flags["aler_on_fast_response"] = str(value)
+        return self
+
+    def disable_translation(self, value: bool = True):
+        self._flags["disable_translation"] = str(value)
+        return self
+
+    def free_text_minimum_characters(self, value: int):
+        self._flags["free_text_minimum_characters"] = str(value)
+        return self
+
+    def no_shuffle(self, value: bool = True):
+        self._flags["no_shuffle"] = str(value)
+        return self
+    
+    def claire_design(self, value: bool = True):
+        self._flags["claire"] = str(value)
+        return self
+    
+    def key_value(self, key: str, value: str):
+        self._flags[key] = value
+        return self

{rapidata-0.1.8 → rapidata-0.1.10}/rapidata/rapidata_client/order/dataset/validation_set_builder.py

@@ -128,6 +128,9 @@ class ValidationSetBuilder:
         for media_path in media_paths:
             if not os.path.exists(media_path):
                 raise FileNotFoundError(f"File not found: {media_path}")
+            
+        # take only last part of truth path
+        truth = os.path.basename(truth)
 
         self._rapid_parts.append(
             ValidatioRapidParts(

{rapidata-0.1.8 → rapidata-0.1.10}/rapidata/rapidata_client/referee/classify_early_stopping_referee.py

@@ -1,4 +1,6 @@
+from typing import Any
 from rapidata.rapidata_client.referee.base_referee import Referee
+from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
 
 
 class ClassifyEarlyStoppingReferee(Referee):
@@ -18,3 +20,10 @@ class ClassifyEarlyStoppingReferee(Referee):
             "threshold": self.threshold,
             "maxVotes": self.max_vote_count,
         }
+    
+    def to_model(self) -> Any:
+        return ProbabilisticAttachCategoryRefereeConfig(
+            _t="ProbabilisticAttachCategoryRefereeConfig",
+            threshold=self.threshold,
+            maxVotes=self.max_vote_count,
+        )

rapidata-0.1.8/rapidata/rapidata_client/feature_flags/feature_flags.py

@@ -1,22 +0,0 @@
-from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
-
-
-class FeatureFlags:
-    def __init__(self):
-        self._flags: dict[str, str] = {}
-
-    def to_list(self) -> list[FeatureFlagModel]:
-        return [FeatureFlagModel(key=name, value=value) for name, value in self._flags.items()]
-    
-    def alert_on_fast_response(self, value: int):
-        self._flags["alertOnFastResponse"] = str(value)
-        return self
-    
-    def disable_translation(self, value: bool):
-        self._flags["disableTranslation"] = str(value)
-        return self
-    
-    def free_text_minimum_characters(self, value: int):
-        self._flags["freeTextMinimumCharacters"] = str(value)
-        return self
-