rapidata 2.40.2__py3-none-any.whl → 2.40.4__py3-none-any.whl

rapidata/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.40.2"
+__version__ = "2.40.4"
 
 from .rapidata_client import (
     RapidataClient,

rapidata/rapidata_client/config/tracer.py

@@ -90,7 +90,6 @@ class RapidataTracer:
                 )
 
                 self._tracer_provider = TracerProvider(resource=resource)
-                trace.set_tracer_provider(self._tracer_provider)
 
                 exporter = OTLPSpanExporter(
                     endpoint="https://otlp-sdk.rapidata.ai/v1/traces",
@@ -100,7 +99,7 @@ class RapidataTracer:
                 span_processor = BatchSpanProcessor(exporter)
                 self._tracer_provider.add_span_processor(span_processor)
 
-                self._real_tracer = trace.get_tracer(self._name)
+                self._real_tracer = self._tracer_provider.get_tracer(self._name)
                 self._otlp_initialized = True
 
             except Exception as e:

rapidata/rapidata_client/order/dataset/_progress_tracker.py

@@ -32,10 +32,11 @@ class ProgressTracker:
             return self.openapi_service.dataset_api.dataset_dataset_id_progress_get(
                 self.dataset_id
             )
-        except Exception:  # noqa: BLE001
+        except Exception:
             return None
 
     def complete(self) -> None:
+        logger.debug("Upload complete, setting upload_complete to True")
         self.upload_complete = True
 
     def run(self) -> None:
@@ -45,6 +46,7 @@ class ProgressTracker:
                 desc="Uploading datapoints",
                 disable=rapidata_config.logging.silent_mode,
             ) as pbar:
+                final_pass = False
                 while True:
                     current_progress = self._get_progress_or_none()
                     if current_progress is None:
@@ -65,6 +67,12 @@ class ProgressTracker:
                         break
 
                     if self.upload_complete and current_progress.pending == 0:
+                        if not final_pass:
+                            logger.debug("Final pass")
+                            time.sleep(self.progress_poll_interval)
+                            final_pass = True
+                            continue
+                        logger.debug("Final pass done, breaking out of loop")
                         break
 
                 pbar.close()
@@ -76,12 +84,13 @@ class ProgressTracker:
                 )
 
                 logger.info(
-                    "Upload complete: %s ready, %s failed (%s%% success rate)",
+                    "Upload complete: %s ready, %s failed, %s pending (%s%% success rate)",
                     current_progress.ready,
                     current_progress.failed,
+                    current_progress.pending,
                     success_rate,
                 )
-        except Exception as e:  # noqa: BLE001
+        except Exception as e:
             logger.error("Progress tracking thread error: %s", str(e))
             raise RuntimeError("Progress tracking failed, aborting uploads")
 

rapidata/rapidata_client/rapidata_client.py

@@ -89,8 +89,6 @@ class RapidataClient:
             logger.debug("Initializing RapidataBenchmarkManager")
             self.mri = RapidataBenchmarkManager(openapi_service=self._openapi_service)
 
-            self._check_beta_features()
-
     def reset_credentials(self):
         """Reset the credentials saved in the configuration file for the current environment."""
         self._openapi_service.reset_credentials()

{rapidata-2.40.2.dist-info → rapidata-2.40.4.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-rapidata/__init__.py,sha256=nNNJT2nQfhHYe5yS9T3V-0MeyarzPFCtE_66Wnit6ho,917
+rapidata/__init__.py,sha256=eYKN-_zLrN6w3N-Mnd0w7MckYzcZC1w_nm9yq-O8xTY,917
 rapidata/api_client/__init__.py,sha256=utY2iWepKJQO_iGz6aIg_qSoqoDkV9pBMAA58pIFE4M,36016
 rapidata/api_client/api/__init__.py,sha256=07qqwzQiBYt5V2BtnzbXhZL2cmVHATyZmCSGshIXLck,1603
 rapidata/api_client/api/benchmark_api.py,sha256=Mlx2qDDJcgPjWvaBnps9dxvVd0re1knG0SyoLUiHKSc,119756
@@ -586,7 +586,7 @@ rapidata/rapidata_client/config/logging_config.py,sha256=ePCh7X6v0Yzkgq1MhFNsHwL
 rapidata/rapidata_client/config/managed_print.py,sha256=2T6dwgR1EZzFAdOEyPp_BBUsa-qrEuhOXgFhsSQKvRo,185
 rapidata/rapidata_client/config/order_config.py,sha256=XxRZERzUUA9md6-PVlV__eCw8DD2kPbT_UmMwG1mAS4,615
 rapidata/rapidata_client/config/rapidata_config.py,sha256=mURnKdl5-2sE4e_IYY9-aBkix6a12t47otEErGE_q0c,1507
-rapidata/rapidata_client/config/tracer.py,sha256=h3GXzaX79HPcip4fBhLaLW0mRlXttR7D3KA78ZT0KVw,4736
+rapidata/rapidata_client/config/tracer.py,sha256=9P27QybtkjJ0U5ocPkIu58DUQB-MgVksQOMhuwKHKcY,4687
 rapidata/rapidata_client/config/upload_config.py,sha256=hjefl-w9WaCNeCEe6hdnrAQEMjgDy-r1zgUUIFR68wk,473
 rapidata/rapidata_client/country_codes/__init__.py,sha256=FB9Dcks44J6C6YBSYmTmNZ71tE130x6NO_3aLJ8fKzQ,40
 rapidata/rapidata_client/country_codes/country_codes.py,sha256=ePHqeb7y9DWQZAnddBzPx1puYBcrgUjdR2sbFijuFD8,283
@@ -632,12 +632,12 @@ rapidata/rapidata_client/filter/response_count_filter.py,sha256=i2u2YQD3_RLQRZyq
 rapidata/rapidata_client/filter/user_score_filter.py,sha256=4B3Zzp7aosDFmte3nLPTlXMN4zatT6Wcq5QLIoXqhgI,1910
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=C-TbKELNuLjQiZt9Gsl6LdtzIUtsu0sNLKGIcLvJEHk,17120
-rapidata/rapidata_client/order/dataset/_progress_tracker.py,sha256=PkTSYrLVNgtXjklXj5ikBEcgF6qaYXoafYyUZQHRn9M,3109
+rapidata/rapidata_client/order/dataset/_progress_tracker.py,sha256=2e9aVMwpdBSO9P3LYn5lygNbtIGzOaJD3b9J7jYaxko,3570
 rapidata/rapidata_client/order/dataset/_rapidata_dataset.py,sha256=ONH56htEvoVZvkSItuTi3_88kaDWohmYYABSoAPEn4Q,10724
 rapidata/rapidata_client/order/rapidata_order.py,sha256=FvZi3t4dARRNsKWvYiNxVvM50AzPwQYR3AzI4utD6OI,14497
 rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=XiV_BpJxG6d8o0rFDYhnB3_mb576CQG5hY-qVXlJZKY,42592
 rapidata/rapidata_client/order/rapidata_results.py,sha256=weL4S14fzug3ZOJbQk9Oj-4tv2jx5aZAMp7VJ-a6Qq4,8437
-rapidata/rapidata_client/rapidata_client.py,sha256=Dm7ZW2tv0QNL1pMk8NHY79vpFXyPeKU6s0WMFHHIH20,5728
+rapidata/rapidata_client/rapidata_client.py,sha256=yEwUsxR0OZbCb2B6QXS23WEVfNZPxRTMi2Hxto2hynw,5687
 rapidata/rapidata_client/referee/__init__.py,sha256=J8oZJNUduPr-Tmn8iJwR-qBiSv7owhUFcEzXTRETecw,155
 rapidata/rapidata_client/referee/_base_referee.py,sha256=aoH3Werw-AtQo2TncG69OmCVvnbXYJBfsRCQry_Pll0,497
 rapidata/rapidata_client/referee/_early_stopping_referee.py,sha256=_JyFcTpncI0PrLf7Ix78FtLto31sdyGDS-2QbVTuvr8,2079
@@ -690,7 +690,7 @@ rapidata/service/credential_manager.py,sha256=T3yL4tXVnibRytxjQkOC-ex3kFGQR5KcKU
 rapidata/service/local_file_service.py,sha256=0Q4LdoEtPFKzgXK2oZ1cQ-X7FipakscjGnnBH8dRFRQ,855
 rapidata/service/openapi_service.py,sha256=k3V4eMNcAjBcxEv17lDivK8LV5TEjRTL9B_5KBlhcas,5482
 rapidata/types/__init__.py,sha256=gSGrmWV5gEA6pPfAR5vwSy_DvibO5IjCZDiB7LtlMOQ,6134
-rapidata-2.40.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.40.2.dist-info/METADATA,sha256=j6edMuFl30ALoelOUbusmBXYPwMCJjfUZbWfzk2GsdY,1406
-rapidata-2.40.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-rapidata-2.40.2.dist-info/RECORD,,
+rapidata-2.40.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.40.4.dist-info/METADATA,sha256=VWYYAZVvVOsnGQ3LytBR37ttKupkFH_Xo9acIWN-IRo,1406
+rapidata-2.40.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+rapidata-2.40.4.dist-info/RECORD,,