rapidata 0.2.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

rapidata/__init__.py

@@ -1 +1,18 @@
-from rapidata.rapidata_client import RapidataClient
+from .rapidata_client import (
+    RapidataClient, 
+    ClassifyWorkflow,
+    TranscriptionWorkflow,
+    CompareWorkflow,
+    FreeTextWorkflow,
+    DemographicSelection,
+    LabelingSelection,
+    ValidationSelection,
+    NaiveReferee,
+    ClassifyEarlyStoppingReferee,
+    PrivateTextMetadata,
+    PublicTextMetadata,
+    PromptMetadata,
+    TranscriptionMetadata,
+    FeatureFlags,
+    CountryCodes
+    )

rapidata/rapidata_client/__init__.py

@@ -1 +1,7 @@
-from .rapidata_client import RapidataClient
+from .rapidata_client import RapidataClient
+from .workflow import ClassifyWorkflow, TranscriptionWorkflow, CompareWorkflow, FreeTextWorkflow
+from .selection import DemographicSelection, LabelingSelection, ValidationSelection
+from .referee import NaiveReferee, ClassifyEarlyStoppingReferee
+from .metadata import PrivateTextMetadata, PublicTextMetadata, PromptMetadata, TranscriptionMetadata
+from .feature_flags import FeatureFlags
+from .country_codes import CountryCodes

rapidata/rapidata_client/country_codes/__init__.py

@@ -1 +1 @@
-from .country_codes import CountryCodes as CountryCodes
+from .country_codes import CountryCodes

rapidata/rapidata_client/feature_flags/__init__.py

@@ -1 +1 @@
-from .feature_flags import FeatureFlags as FeatureFlags
+from .feature_flags import FeatureFlags

rapidata/rapidata_client/metadata/__init__.py

@@ -0,0 +1,5 @@
+from .base_metadata import Metadata
+from .private_text_metadata import PrivateTextMetadata
+from .public_text_metadata import PublicTextMetadata
+from .prompt_metadata import PromptMetadata
+from .transcription_metadata import TranscriptionMetadata

rapidata/rapidata_client/metadata/public_text_metadata.py

@@ -0,0 +1,16 @@
+from rapidata.api_client.models.public_text_metadata_input import (
+    PublicTextMetadataInput,
+)
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
+
+
+class PublicTextMetadata(Metadata):
+
+    def __init__(self, text: str, identifier: str = "public_text"):
+        super().__init__(identifier=identifier)
+        self._text = text
+
+    def to_model(self):
+        return PublicTextMetadataInput(
+            _t="PublicTextMetadataInput", identifier=self._identifier, text=self._text
+        )

rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -23,6 +23,8 @@ from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
 from rapidata.rapidata_client.referee import Referee
 from rapidata.service.openapi_service import OpenAPIService
 
+from rapidata.rapidata_client.workflow.compare_workflow import CompareWorkflow
+
 
 class RapidataOrderBuilder:
     """Builder object for creating Rapidata orders.
@@ -51,10 +53,11 @@ class RapidataOrderBuilder:
         self._country_codes: list[str] | None = None
         self._selections: list[Selection] = []
         self._rapids_per_bag: int = 2
+        self._priority: int = 50
 
     def _to_model(self) -> CreateOrderModel:
         if self._workflow is None:
-            raise ValueError("You must provide a blueprint to create an order.")
+            raise ValueError("You must provide a workflow to create an order.")
 
         if self._referee is None:
             print("No referee provided, using default NaiveReferee.")
@@ -85,6 +88,7 @@ class RapidataOrderBuilder:
                 CreateOrderModelSelectionsInner(selection.to_model())
                 for selection in self._selections
             ],
+            priority=self._priority,
         )
 
     def create(self, submit=True, max_workers=10) -> RapidataOrder:
@@ -100,6 +104,8 @@ class RapidataOrderBuilder:
             ValueError: If no workflow is provided.
         """
         order_model = self._to_model()
+        if isinstance(self._workflow, CompareWorkflow): # temp fix, will be handeled by backend in the future
+            assert all([len(path) == 2 for path in self._media_paths]), "The media paths must come in pairs for comparison tasks."
 
         result = self._openapi_service.order_api.order_create_post(
             create_order_model=order_model
@@ -114,7 +120,6 @@ class RapidataOrderBuilder:
         )
 
         order.dataset.add_media_from_paths(self._media_paths, self._metadata, max_workers)
-
         if submit:
             order.submit()
 
@@ -235,3 +240,15 @@ class RapidataOrderBuilder:
         """
         self._selections = selections
         return self
+    
+    def priority(self, priority: int):
+        """Set the priority for the order.
+
+        Args:
+            priority (int): The priority to be set.
+
+        Returns:
+            RapidataOrderBuilder: The updated RapidataOrderBuilder instance.
+        """
+        self._priority = priority
+        return self

rapidata/rapidata_client/rapidata_client.py

@@ -5,6 +5,9 @@ from rapidata.rapidata_client.dataset.validation_set_builder import ValidationSe
 from rapidata.rapidata_client.order.rapidata_order_builder import RapidataOrderBuilder
 from rapidata.rapidata_client.utils.utils import Utils
 from rapidata.service.openapi_service import OpenAPIService
+from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
+from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
+
 
 
 class RapidataClient:
@@ -59,6 +62,25 @@ class RapidataClient:
             RapidataValidationSet: The ValidationSet instance.
         """
         return RapidataValidationSet(validation_set_id, self.openapi_service)
+    
+    def get_order(self, order_id: str) -> RapidataOrder:
+        """Get an order by ID.
+
+        Args:
+            order_id (str): The ID of the order.
+
+        Returns:
+            RapidataOrder: The Order instance.
+        """
+
+        # TODO: check the pipeline for the dataset id - not really necessary atm
+        # order = self.openapi_service.order_api.order_get_by_id_get(order_id)
+        # pipeline = self.openapi_service..pipeline_get_by_id_get(order.pipeline_id)
+        temp_dataset = RapidataDataset("temp", self.openapi_service)
+        return RapidataOrder(
+            dataset=temp_dataset, 
+            order_id=order_id, 
+            openapi_service=self.openapi_service)
 
     @property
     def utils(self) -> Utils:

rapidata/rapidata_client/referee/__init__.py

@@ -1,3 +1,3 @@
-from .base_referee import Referee as Referee
-from .naive_referee import NaiveReferee as NaiveReferee
-from .classify_early_stopping_referee import ClassifyEarlyStoppingReferee as ClassifyEarlyStoppingReferee
+from .base_referee import Referee
+from .naive_referee import NaiveReferee #as MaxVoteReferee
+from .classify_early_stopping_referee import ClassifyEarlyStoppingReferee

rapidata/rapidata_client/selection/__init__.py

@@ -0,0 +1,4 @@
+from .base_selection import Selection
+from .demographic_selection import DemographicSelection
+from .labeling_selection import LabelingSelection
+from .validation_selection import ValidationSelection

rapidata/rapidata_client/workflow/__init__.py

@@ -1,4 +1,5 @@
-from .base_workflow import Workflow as Workflow
-from .classify_workflow import ClassifyWorkflow as ClassifyWorkflow
-from .compare_workflow import CompareWorkflow as CompareWorkflow
-from .free_text_workflow import FreeTextWorkflow as FreeTextWorkflow
+from .base_workflow import Workflow 
+from .classify_workflow import ClassifyWorkflow
+from .compare_workflow import CompareWorkflow
+from .free_text_workflow import FreeTextWorkflow
+from .transcription_workflow import TranscriptionWorkflow

{rapidata-0.2.1.dist-info → rapidata-0.4.0.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-rapidata/__init__.py,sha256=iUiqnIukWZ4ygW9_TNt7hn1ZnDROoi6vlS5MfJq0OE4,51
+rapidata/__init__.py,sha256=mawPp5WoEVziUnaycojmzCyUDMgcIcUXqGEv2dGgDtk,407
 rapidata/api_client/__init__.py,sha256=eLFuUpHlUktj4p6EyWzC9H4Fp-asEoM6q3SPHOI2FJA,18786
 rapidata/api_client/api/__init__.py,sha256=NmEdKcHiQlR8hn7KMT-V5CZv9H-uPlR8ZHJR27gUq2g,524
 rapidata/api_client/api/coco_api.py,sha256=v5xYyQVlAv4ENgihWlji3BkbczQYKxLP0gnxfQLRmkI,24926
@@ -220,38 +220,39 @@ rapidata/api_client/models/workflow_split_model.py,sha256=zthOSaUl8dbLhLymLK_lrP
 rapidata/api_client/models/workflow_split_model_filter_configs_inner.py,sha256=1Fx9uZtztiiAdMXkj7YeCqt7o6VkG9lKf7D7UP_h088,7447
 rapidata/api_client/rest.py,sha256=A8BNcRsSo-Pn6sluvAOkTONtiA8gpulLjvGX93sQiAs,9419
 rapidata/api_client_README.md,sha256=ebeOypm4MabEycVUxBm8YQjjijLaSJMmxtlA3cFklP4,30117
-rapidata/rapidata_client/__init__.py,sha256=JUnCWjIxuAROIKid96Fz-Kiul7PmtnNofzwhTclAU2I,43
+rapidata/rapidata_client/__init__.py,sha256=99DP3H19vIvmXqqNUjnoT-8aaZdtxCaOOgP8yYSbVB4,470
 rapidata/rapidata_client/config.py,sha256=tQLgN6k_ATOX1GzZh38At2rgBDLStV6rJ6z0vsTTPjg,186
-rapidata/rapidata_client/country_codes/__init__.py,sha256=Y8qeG2IMjvMGvhaPydq0nhwRQHb6dQqilctlEXu0_PE,55
+rapidata/rapidata_client/country_codes/__init__.py,sha256=FB9Dcks44J6C6YBSYmTmNZ71tE130x6NO_3aLJ8fKzQ,40
 rapidata/rapidata_client/country_codes/country_codes.py,sha256=Q0HMX7uHJQDeLCFPP5bq4iYi6pgcDWEcl2ONGhjgoeU,286
 rapidata/rapidata_client/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/dataset/rapidata_dataset.py,sha256=QDsl7ZCZxuG02yfEpBSphfSDZh_qHz6m5HUCGwZflfw,3205
 rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=YrnUzia9AXgq2z917FtztFxj4fD5EgTWXVPBzLVIujY,9374
 rapidata/rapidata_client/dataset/validation_rapid_parts.py,sha256=SIeQesEXPPOW5kclxYLNWaKllBXHm7DQKBdMU-GXnfc,2104
 rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=B9D-uNCo_PO0NCUHju_7dsWtz_KcOmvFIsxUgQ67Q2M,7471
-rapidata/rapidata_client/feature_flags/__init__.py,sha256=BNG_NQ4CrrC61fAWliImr8r581pIvegrkepVVbxcBw8,55
+rapidata/rapidata_client/feature_flags/__init__.py,sha256=IYkcK_bZCl5RfyQFiWjjUdz4y0jipiW9qfeopq4EjQQ,40
 rapidata/rapidata_client/feature_flags/feature_flags.py,sha256=hcS9YRzpsPWpZfw-3QwSuf2TaVg-MOHBxY788oNqIW4,3957
-rapidata/rapidata_client/metadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rapidata/rapidata_client/metadata/__init__.py,sha256=qMmo4wqScUCAJ6YXRWxvJLmbFA5YRbK39p9_exV1d50,246
 rapidata/rapidata_client/metadata/base_metadata.py,sha256=BJoZT4lRs94lu7PE3__LcsMINHNDVKRKRART-EKzSaw,217
 rapidata/rapidata_client/metadata/private_text_metadata.py,sha256=VJ_SJwp8OMFZLuUMRWvPeKOySXYJuGr-YgpwuLfTbmo,517
 rapidata/rapidata_client/metadata/prompt_metadata.py,sha256=_FypjKWrC3iKUO_G2CVwAGcYbEYClVv4K7upqlnmecw,468
+rapidata/rapidata_client/metadata/public_text_metadata.py,sha256=LTiBQHs6izxQ6-C84d6Pf7lL4ENTDgg__HZnDKvzvMc,511
 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=DCfw_xH7Ja8KURlJYaxVuX6dM81sUF7HLyX4iTfqqZc,3098
-rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=Wx7lhThTd6SwJNSbXzhGCsEgMhizjsriGz2zMjbQyEI,8134
-rapidata/rapidata_client/rapidata_client.py,sha256=z3vz5_GNivnShj7kqii-eUff16rvwSy62zwi8WZqAWo,2776
-rapidata/rapidata_client/referee/__init__.py,sha256=x0AxGCsR6TlDjfqQ00lB9V7QVS9EZCJzweNEIzx42PI,207
+rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=lciogpvXZhMNe8RE_6qLtHNSzSQobfqOXdw106f9mzg,8838
+rapidata/rapidata_client/rapidata_client.py,sha256=NsLE6WRrnPFNyCklD0d249XmCoVr_Z_z8HOf-scEnxM,3623
+rapidata/rapidata_client/referee/__init__.py,sha256=Ow9MQsONhF4sX2wFK9jbvSBrpcJgtq3OglIQMkBUdIY,167
 rapidata/rapidata_client/referee/base_referee.py,sha256=bMy7cw0a-pGNbFu6u_1_Jplu0A483Ubj4oDQzh8vu8k,493
 rapidata/rapidata_client/referee/classify_early_stopping_referee.py,sha256=B5wsqKM3_Oc1TU_MFGiIyiXjwK1LcmaVjhzLdaL8Cgw,1797
 rapidata/rapidata_client/referee/naive_referee.py,sha256=KWMLSc73gOdM8YT_ciFhfN7J4eKgtOFphBG9tIra9g0,1179
-rapidata/rapidata_client/selection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rapidata/rapidata_client/selection/__init__.py,sha256=KogO0_Hc0ENP5YWZrWhJ5AbNURJvmGuJurFrOuNyjps,198
 rapidata/rapidata_client/selection/base_selection.py,sha256=Y3HkROPm4I4HLNiR0HuHKpvk236KkRlsoDxQATm_chY,138
 rapidata/rapidata_client/selection/demographic_selection.py,sha256=DU8YvAj-WFALMAQJgjvBm1_ruIdPg2JVuKw07f7nDuA,453
 rapidata/rapidata_client/selection/labeling_selection.py,sha256=cqDMQEXfQGMmgIvPgGOYgIGaXflV_J7LZsGOsakLXqo,425
 rapidata/rapidata_client/selection/validation_selection.py,sha256=HswzD2SvZZWisNLoGj--0sT_TIK8crYp3xGGndo6aLY,523
 rapidata/rapidata_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/utils/utils.py,sha256=Fl99gCnh_HnieIp099xEvEv4g2kEIKiFcUp0G2iz6x8,815
-rapidata/rapidata_client/workflow/__init__.py,sha256=CDG8bKOBhLV4A0uBlqOd__79fH9KW3-UlZsR1RANLf0,250
+rapidata/rapidata_client/workflow/__init__.py,sha256=nHB6heVVf_A3nhSL0NGapnGqJAL0K9nfOpfyaUM5srw,238
 rapidata/rapidata_client/workflow/base_workflow.py,sha256=VoZtNIFoylTrylMWC7CqzuSCLfG2yEGCmenf-HBzYOI,1196
 rapidata/rapidata_client/workflow/classify_workflow.py,sha256=NkyyBrlCDqYVQaCARR9EHYuREEtXond69kD6jbzcN3M,1713
 rapidata/rapidata_client/workflow/compare_workflow.py,sha256=nqrgtohlXBfgHh_cRPOFAlwdI8wVc7PUsH0FDM7wIjg,1431
@@ -262,7 +263,7 @@ rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5
 rapidata/service/openapi_service.py,sha256=-vrM2jEzQxr9KAerOYkVhpvMEeHwjzRwm9L_VFyzOT0,1537
 rapidata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/utils/image_utils.py,sha256=TldO3eJWG8IhfJjm5MfNGO0mEDm1mQTsRoA0HLU1Uxs,404
-rapidata-0.2.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-0.2.1.dist-info/METADATA,sha256=3Uw9M9uj4-Za3qasR5n2w6LITe8O21Ladaqs0XBpgos,961
-rapidata-0.2.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rapidata-0.2.1.dist-info/RECORD,,
+rapidata-0.4.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-0.4.0.dist-info/METADATA,sha256=F6QuVl_gSq63EK07SgYL97-63aTgaZQBUKVB7z___L0,961
+rapidata-0.4.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rapidata-0.4.0.dist-info/RECORD,,