rapidata 2.1.3__py3-none-any.whl → 2.1.4__py3-none-any.whl

rapidata/rapidata_client/order/rapidata_order.py

@@ -3,6 +3,7 @@ from rapidata.rapidata_client.order._rapidata_dataset import RapidataDataset
 from rapidata.service.openapi_service import OpenAPIService
 import json
 from rapidata.api_client.exceptions import ApiException
+from rapidata.api_client.models.order_state import OrderState
 from typing import Optional, cast, Any
 from rapidata.api_client.models.workflow_artifact_model import WorkflowArtifactModel
 from tqdm import tqdm
@@ -78,6 +79,17 @@ class RapidataOrder:
         if refresh_rate < 1:
             raise ValueError("refresh_rate must be at least 1")
         
+        if self.get_status() == OrderState.CREATED:
+            raise Exception("Order has not been started yet. Please start it first.")
+        
+        while self.get_status() == OrderState.SUBMITTED:
+            print(f"Order '{self.name}' is submitted and being reviewed. standby...", end="\r")
+            sleep(1)
+
+        if self.get_status() == OrderState.MANUALREVIEW:
+            raise Exception(f"Order '{self.name}' is in manual review. It might take some time to start. To speed up the process, please contact support (info@rapidata.ai).\
+                            \nOnce the order has started, you can run this method again to display the progress bar.")
+
         with tqdm(total=100, desc="Processing order", unit="%", bar_format="{desc}: {percentage:3.0f}%|{bar}| completed [{elapsed}<{remaining}, {rate_fmt}]") as pbar:
             last_percentage = 0
             while True:
@@ -95,7 +107,7 @@ class RapidataOrder:
         if self.__workflow_id:
             return self.__workflow_id
 
-        for _ in range(10):
+        for _ in range(10): # Try for 20 seconds to get the workflow id if workflow has not started by then, raise an exception
             try:
                 order_result = self.__openapi_service.order_api.order_get_by_id_get(self.order_id)
                 pipeline = self.__openapi_service.pipeline_api.pipeline_id_get(order_result.pipeline_id)
@@ -104,7 +116,7 @@ class RapidataOrder:
             except Exception:
                 sleep(2)
         if not self.__workflow_id:
-            raise Exception("Order has not started yet. Please start it or wait for a few seconds and try again.")
+            raise Exception("Something went wrong when trying to get the order progress.")
         return self.__workflow_id
     
     def __get_workflow_progress(self):
@@ -131,7 +143,7 @@ class RapidataOrder:
         Returns: 
             The results of the order.
         """
-        while self.get_status() not in ["Completed", "Paused", "ManuelReview", "Failed"]:
+        while self.get_status() not in [OrderState.COMPLETED, OrderState.PAUSED, OrderState.MANUALREVIEW, OrderState.FAILED]:
             sleep(5)
 
         try:

{rapidata-2.1.3.dist-info → rapidata-2.1.4.dist-info}/METADATA

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

@@ -368,7 +368,7 @@ rapidata/rapidata_client/metadata/_select_words_metadata.py,sha256=I4qVtCkj60ljk
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/order/_rapidata_dataset.py,sha256=IU0N2_LksyEDlvLq56vCkiSR45kA9I6q4pqDoEpWmMw,5240
 rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=R4iOWy8nfIHpzqqqQcQhLgpdB8z_tUrkL2M_8rZUw9M,13121
-rapidata/rapidata_client/order/rapidata_order.py,sha256=e7ZMllbOaBFqPDVApL-kemAnFLbzF9rNWyTGOKuU8Qc,5741
+rapidata/rapidata_client/order/rapidata_order.py,sha256=ti5fHd-KAvI_WRwe9UWF6QDx-VYzlU_Kqmi44wphJzc,6588
 rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=ZTygrv5x7dWjWxXOgFcdtyei_ap6J33SJWuWB_AwXeg,24591
 rapidata/rapidata_client/rapidata_client.py,sha256=9SMOouDBwe-aIi9rnPtnL2vrqnlQHDKxOUpw-Eh3Fsg,1782
 rapidata/rapidata_client/referee/__init__.py,sha256=q0Hv9nmfEpyChejtyMLT8hWKL0vTTf_UgUXPYNJ-H6M,153
@@ -417,7 +417,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.1.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.1.3.dist-info/METADATA,sha256=qtAlYurcVv92pq9I3LzFMppkg2RpmJAGg7CruP5rW50,1067
-rapidata-2.1.3.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-rapidata-2.1.3.dist-info/RECORD,,
+rapidata-2.1.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.1.4.dist-info/METADATA,sha256=RuxhTz67gYAWeeYsbrLVADaNb91T2DMMd6RtVaeprH8,1067
+rapidata-2.1.4.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+rapidata-2.1.4.dist-info/RECORD,,