rapidata 2.4.1__py3-none-any.whl → 2.5.0__py3-none-any.whl

rapidata/rapidata_client/validation/rapidata_validation_set.py

@@ -1,3 +1,6 @@
+from rapidata.rapidata_client.validation.rapids.rapids import Rapid
+from rapidata.service.openapi_service import OpenAPIService
+
 class RapidataValidationSet:
     """A class for interacting with a Rapidata validation set.
 
@@ -10,9 +13,18 @@ class RapidataValidationSet:
         name (str): The name of the validation set.
     """
 
-    def __init__(self, validation_set_id, name: str):
+    def __init__(self, validation_set_id, name: str, openapi_service: OpenAPIService):
         self.id = validation_set_id
         self.name = name
+        self.__openapi_service = openapi_service
+    
+    def add_rapid(self, rapid: Rapid):
+        """Add a Rapid to the validation set.
+
+        Args:
+            rapid (Rapid): The Rapid to add to the validation set.
+        """
+        rapid._add_to_validation_set(self.id, self.__openapi_service)
 
     def __str__(self):
         return f"name: '{self.name}' id: {self.id}"

rapidata/rapidata_client/validation/rapids/rapids.py

@@ -20,6 +20,7 @@ from rapidata.api_client.models.datapoint_metadata_model_metadata_inner import (
     DatapointMetadataModelMetadataInner,
 )
 
+from rapidata.service.openapi_service import OpenAPIService
 
 class Rapid():
     def __init__(self, asset: MediaAsset | TextAsset | MultiAsset, metadata: Sequence[Metadata], payload: Any, truth: Any, randomCorrectProbability: float, explanation: str | None):
@@ -30,7 +31,22 @@ class Rapid():
         self.randomCorrectProbability = randomCorrectProbability
         self.explanation = explanation 
 
-    def to_media_model(self, validationSetId: str) -> tuple[AddValidationRapidModel, list[StrictStr | tuple[StrictStr, StrictBytes] | StrictBytes]]:
+    def _add_to_validation_set(self, validationSetId: str, openapi_service: OpenAPIService):
+        if isinstance(self.asset, TextAsset) or (isinstance(self.asset, MultiAsset) and isinstance(self.asset.assets[0], TextAsset)):
+            openapi_service.validation_api.validation_add_validation_text_rapid_post(
+                add_validation_text_rapid_model=self.__to_text_model(validationSetId)
+            )
+
+        elif isinstance(self.asset, MediaAsset) or (isinstance(self.asset, MultiAsset) and isinstance(self.asset.assets[0], MediaAsset)):
+            model = self.__to_media_model(validationSetId)
+            openapi_service.validation_api.validation_add_validation_rapid_post(
+                model=model[0], files=model[1]
+            )
+            
+        else:
+            raise TypeError("The asset must be a MediaAsset, TextAsset, or MultiAsset")
+
+    def __to_media_model(self, validationSetId: str) -> tuple[AddValidationRapidModel, list[StrictStr | tuple[StrictStr, StrictBytes] | StrictBytes]]:
         assets: list[MediaAsset] = [] 
         if isinstance(self.asset, MultiAsset):
             for asset in self.asset.assets:
@@ -57,7 +73,7 @@ class Rapid():
             explanation=self.explanation
         ), [asset.to_file() for asset in assets])
 
-    def to_text_model(self, validationSetId: str) -> AddValidationTextRapidModel:
+    def __to_text_model(self, validationSetId: str) -> AddValidationTextRapidModel:
         texts: list[str] = []
         if isinstance(self.asset, MultiAsset):
             for asset in self.asset.assets:

rapidata/rapidata_client/validation/rapids/rapids_manager.py

@@ -132,7 +132,7 @@ class RapidsManager:
         asset = MediaAsset(datapoint)
         transcription_words = [
             TranscriptionWord(word=word, wordIndex=i)
-            for i, word in enumerate(sentence)
+            for i, word in enumerate(sentence.split(" "))
         ]
 
         correct_transcription_words: list[TranscriptionWord] = []

rapidata/rapidata_client/validation/validation_set_manager.py

@@ -85,7 +85,7 @@ class ValidationSetManager:
                 )
             )
 
-        return self._submit(name=name, data_type=data_type, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
 
     def create_compare_set(self,
         name: str,
@@ -142,7 +142,7 @@ class ValidationSetManager:
                 )
             )
  
-        return self._submit(name=name, data_type=data_type, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
   
     def create_select_words_set(self,
         name: str,
@@ -197,7 +197,7 @@ class ValidationSetManager:
                 )
             )
 
-        return self._submit(name=name, data_type=RapidataDataTypes.MEDIA, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
 
     def create_locate_set(self,
         name: str,
@@ -249,7 +249,7 @@ class ValidationSetManager:
                 )
             )
         
-        return self._submit(name=name, data_type=RapidataDataTypes.MEDIA, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
     
     def create_draw_set(self,
         name: str,
@@ -300,7 +300,7 @@ class ValidationSetManager:
                 )
             )
 
-        return self._submit(name=name, data_type=RapidataDataTypes.MEDIA, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
 
     def create_timestamp_set(self,
         name: str,
@@ -352,11 +352,10 @@ class ValidationSetManager:
                 )
             )
 
-        return self._submit(name=name, data_type=RapidataDataTypes.MEDIA, rapids=rapids, print_confirmation=print_confirmation)
+        return self._submit(name=name, rapids=rapids, print_confirmation=print_confirmation)
     
     def create_mixed_set(self,
         name: str,
-        data_type: str,
         rapids: list[Rapid],
         print_confirmation: bool = True
     ) -> RapidataValidationSet:
@@ -368,7 +367,7 @@ class ValidationSetManager:
             print_confirmation (bool, optional): Whether to print a confirmation message that validation set has been created. Defaults to True.
         """
 
-        return self._submit(name, data_type, rapids, print_confirmation)
+        return self._submit(name, rapids, print_confirmation)
     
     def get_validation_set_by_id(self, validation_set_id: str) -> RapidataValidationSet:
         """Get a validation set by ID.
@@ -384,9 +383,9 @@ class ValidationSetManager:
         except Exception:
             raise ValueError(f"ValidationSet with ID {validation_set_id} not found.")
         
-        return RapidataValidationSet(validation_set_id, validation_set.name)
+        return RapidataValidationSet(validation_set_id, validation_set.name, self.__openapi_service)
 
-    def _submit(self, name: str, data_type: str, rapids: list[Rapid], print_confirmation: bool) -> RapidataValidationSet:
+    def _submit(self, name: str, rapids: list[Rapid], print_confirmation: bool) -> RapidataValidationSet:
         validation_set_id = (
             self.__openapi_service.validation_api.validation_create_validation_set_post(
                 name=name
@@ -399,22 +398,17 @@ class ValidationSetManager:
         if print_confirmation:
             print(f"Validation set '{name}' created with ID {validation_set_id}")
 
-        for rapid in rapids:
-            if data_type == RapidataDataTypes.TEXT:
-                self.__openapi_service.validation_api.validation_add_validation_text_rapid_post(
-                    add_validation_text_rapid_model=rapid.to_text_model(validation_set_id)
-                )
-            else:
-                model = rapid.to_media_model(validation_set_id)
-                self.__openapi_service.validation_api.validation_add_validation_rapid_post(
-                   model=model[0], files=model[1]
-                )
-
-        return RapidataValidationSet(
+        validation_set = RapidataValidationSet(
             name=name,
             validation_set_id=validation_set_id,
+            openapi_service=self.__openapi_service
         )
 
+        for rapid in rapids:
+            validation_set.add_rapid(rapid)
+        
+        return validation_set
+
 
     def find_validation_sets(self, name: str = "", amount: int = 1) -> list[RapidataValidationSet]:
         """Find validation sets by name.

{rapidata-2.4.1.dist-info → rapidata-2.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rapidata
-Version: 2.4.1
+Version: 2.5.0
 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-2.4.1.dist-info → rapidata-2.5.0.dist-info}/RECORD

@@ -406,12 +406,12 @@ rapidata/rapidata_client/settings/play_video_until_the_end.py,sha256=LLHx2_72k5Z
 rapidata/rapidata_client/settings/rapidata_settings.py,sha256=Kjxm4GStrpfLKylx84BiKdEQOjvNfO74DiSChmH63fg,1165
 rapidata/rapidata_client/settings/translation_behaviour.py,sha256=i9n_H0eKJyKW6m3MKH_Cm1XEKWVEWsAV_79xGmGIC-4,742
 rapidata/rapidata_client/validation/__init__.py,sha256=s5wHVtcJkncXSFuL9I0zNwccNOKpWAqxqUjkeohzi2E,24
-rapidata/rapidata_client/validation/rapidata_validation_set.py,sha256=c4ywPef1yGwx0D9wTzj7465p-VQa1V6WnHJUU5ReK7Q,646
+rapidata/rapidata_client/validation/rapidata_validation_set.py,sha256=IkHOdQX3ihBBHDvG2EE6wu1ZSG7evOWMYyOx9Lva0tw,1111
 rapidata/rapidata_client/validation/rapids/__init__.py,sha256=WU5PPwtTJlte6U90MDakzx4I8Y0laj7siw9teeXj5R0,21
 rapidata/rapidata_client/validation/rapids/box.py,sha256=t3_Kn6doKXdnJdtbwefXnYKPiTKHneJl9E2inkDSqL8,589
-rapidata/rapidata_client/validation/rapids/rapids.py,sha256=sZEzKj4FdVChm_rUwcmG_v7X-yO5Eh4q1g54T4c4puc,3507
-rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=rDmxYkv067O-S95TNc789FktfCPBCudyzBRLSArmfus,14236
-rapidata/rapidata_client/validation/validation_set_manager.py,sha256=I2ICynEOjgmp5KDUceXCKGyKGSt21N_GTn1Mi4-6DTc,23295
+rapidata/rapidata_client/validation/rapids/rapids.py,sha256=aWkcjWR6Pr2BFwz8a0MfqEkXnEJPkAbKMeQtGXnsags,4440
+rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=4HOX6c42sCOfpGR2aDoiMgrfjontM8z2KEJwQp4ir6A,14247
+rapidata/rapidata_client/validation/validation_set_manager.py,sha256=RR3lP3VtOVCtmbKjEAFYVbQxZyUCXvOuWMqw8dGz-WA,22730
 rapidata/rapidata_client/workflow/__init__.py,sha256=eFRx0fm280alXpds6hYcnxN_yERlabF9B5sTdPFsL1g,430
 rapidata/rapidata_client/workflow/_base_workflow.py,sha256=XyIZFKS_RxAuwIHS848S3AyLEHqd07oTD_5jm2oUbsw,762
 rapidata/rapidata_client/workflow/_classify_workflow.py,sha256=9bT54wxVJgxC-zLk6MVNbseFpzYrvFPjt7DHvxqYfnk,1736
@@ -427,7 +427,7 @@ rapidata/service/credential_manager.py,sha256=Of0BQs_V1T7rkrWX9groLX790nOknaARwn
 rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5AnfY14BW8,877
 rapidata/service/openapi_service.py,sha256=Z4NrAuilLlIWBdGOv6otz36tHS_vvU36w5jmvOUTmqo,3198
 rapidata/service/token_manager.py,sha256=JZ5YbR5Di8dO3H4kK11d0kzWlrXxjgCmeNkHA4AapCM,6425
-rapidata-2.4.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.4.1.dist-info/METADATA,sha256=nOQUJTI2cPvoUj7WS7N1Y2d5L_ZW1Tx81tb8SawafqQ,1107
-rapidata-2.4.1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-rapidata-2.4.1.dist-info/RECORD,,
+rapidata-2.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.5.0.dist-info/METADATA,sha256=Bi3-kOaNpfnjqHS9QwC0ua5dnnlhluiMZdopp0fx08U,1107
+rapidata-2.5.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+rapidata-2.5.0.dist-info/RECORD,,