upgini 1.1.132__py3-none-any.whl → 1.1.132a1__py3-none-any.whl

upgini/http.py

@@ -32,7 +32,7 @@ from upgini.metadata import (
     SearchCustomization,
 )
 from upgini.resource_bundle import bundle
-from upgini.utils.track_info import get_track_metrics_with_timeout
+from upgini.utils.track_info import get_track_metrics
 
 try:
     from importlib_metadata import version
@@ -49,7 +49,6 @@ except ImportError:
 UPGINI_URL: str = "UPGINI_URL"
 UPGINI_API_KEY: str = "UPGINI_API_KEY"
 DEMO_API_KEY: str = "Aa4BPwGFbn1zNEXIkZ-NbhsRk0ricN6puKuga1-O5lM"
-TRACK_METRICS_TIMEOUT_SECONDS: int = 10
 
 refresh_token_lock = threading.Lock()
 
@@ -356,7 +355,9 @@ class _RestClient:
                 digest = md5_hash.hexdigest()
                 metadata_with_md5 = metadata.copy(update={"checksumMD5": digest})
 
-            digest_sha256 = hashlib.sha256(pd.util.hash_pandas_object(pd.read_parquet(file_path)).values).hexdigest()
+            digest_sha256 = hashlib.sha256(
+                pd.util.hash_pandas_object(pd.read_parquet(file_path)).values
+            ).hexdigest()
             metadata_with_md5 = metadata_with_md5.copy(update={"digest": digest_sha256})
 
             with open(file_path, "rb") as file:
@@ -375,11 +376,7 @@ class _RestClient:
                         search_customization.json(exclude_none=True).encode(),
                         "application/json",
                     )
-                files["tracking"] = (
-                    "tracking.json",
-                    dumps(get_track_metrics_with_timeout(TRACK_METRICS_TIMEOUT_SECONDS)).encode(),
-                    "application/json",
-                )
+                files["tracking"] = ("tracking.json", dumps(get_track_metrics()).encode(), "application/json")
                 additional_headers = {self.SEARCH_KEYS_HEADER_NAME: ",".join(self.search_keys_meaning_types(metadata))}
 
                 return self._send_post_file_req_v2(
@@ -438,7 +435,9 @@ class _RestClient:
                 digest = md5_hash.hexdigest()
                 metadata_with_md5 = metadata.copy(update={"checksumMD5": digest})
 
-            digest_sha256 = hashlib.sha256(pd.util.hash_pandas_object(pd.read_parquet(file_path)).values).hexdigest()
+            digest_sha256 = hashlib.sha256(
+                pd.util.hash_pandas_object(pd.read_parquet(file_path)).values
+            ).hexdigest()
             metadata_with_md5 = metadata_with_md5.copy(update={"digest": digest_sha256})
 
             with open(file_path, "rb") as file:
@@ -457,11 +456,7 @@ class _RestClient:
                         search_customization.json(exclude_none=True).encode(),
                         "application/json",
                     )
-                files["tracking"] = (
-                    "ide",
-                    dumps(get_track_metrics_with_timeout(TRACK_METRICS_TIMEOUT_SECONDS)).encode(),
-                    "application/json",
-                )
+                files["tracking"] = ("ide", dumps(get_track_metrics()).encode(), "application/json")
 
                 additional_headers = {self.SEARCH_KEYS_HEADER_NAME: ",".join(self.search_keys_meaning_types(metadata))}
 
@@ -802,7 +797,7 @@ class BackendLogHandler(logging.Handler):
         def task():
             try:
                 if self.track_metrics is None:
-                    self.track_metrics = get_track_metrics_with_timeout(TRACK_METRICS_TIMEOUT_SECONDS)
+                    self.track_metrics = get_track_metrics()
                     if "ip" in self.track_metrics.keys():
                         self.hostname = self.track_metrics["ip"]
                     else:

upgini/utils/track_info.py

@@ -6,7 +6,6 @@ from functools import lru_cache
 from getpass import getuser
 from hashlib import sha256
 from uuid import getnode
-from concurrent import futures
 
 from requests import get, post
 
@@ -46,17 +45,6 @@ def _get_execution_ide() -> str:
         return "other"
 
 
-def get_track_metrics_with_timeout(timeout_seconds: int = 10) -> dict:
-    with futures.ThreadPoolExecutor() as executor:
-        future = executor.submit(get_track_metrics)
-        try:
-            result = future.result(timeout_seconds)
-            return result
-        except futures.TimeoutError:
-            future.cancel()
-            return dict()
-
-
 @lru_cache()
 def get_track_metrics() -> dict:
     # default values

{upgini-1.1.132.dist-info → upgini-1.1.132a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: upgini
-Version: 1.1.132
+Version: 1.1.132a1
 Summary: Intelligent data search & enrichment for Machine Learning
 Home-page: https://upgini.com/
 Author: Upgini Developers

{upgini-1.1.132.dist-info → upgini-1.1.132a1.dist-info}/RECORD

@@ -3,7 +3,7 @@ upgini/ads.py,sha256=mre6xn44wcC_fg63iLT_kTh4mViZqR9AKRJZAtpQz8Y,2592
 upgini/dataset.py,sha256=uiba8qC2RNvqka5MQp-XgdYDE7-nqO5FojMO5a0n_HA,42978
 upgini/errors.py,sha256=BqpvfhW2jJW5fa5KXj0alhXatGl-WK4xTl309-QNLp8,959
 upgini/features_enricher.py,sha256=iV9OHy8QFQdB5uUX6S2RuI28WBZFNkR5LZwRrhuB8Oo,118746
-upgini/http.py,sha256=164ysTRGZ2hRtsFTjl4vBsyaz0iS75jiF7BYGFk2MVc,36468
+upgini/http.py,sha256=rY9kRwqErjC8qewANiBhkPObek-Hrm9037KEyQ3vpfs,36191
 upgini/metadata.py,sha256=Oefg-rkA4PsZUHIho_clZcnyZwdtVJ1gXPvEY6oBmpg,5969
 upgini/metrics.py,sha256=5tQ_6ZKUM0EBLnHmKZD63KHCQXTYGeE-uFLs0wHcYf4,15477
 upgini/search_task.py,sha256=cM-3cAcdnKHrCl7ZTqhHLrSkDVpdgowV5YBaFq2BzjY,13597
@@ -37,10 +37,10 @@ upgini/utils/format.py,sha256=Yv5cvvSs2bOLUzzNu96Pu33VMDNbabio92QepUj41jU,243
 upgini/utils/phone_utils.py,sha256=JNSkF8G6mgsN8Czy11pamaJdsY6rBINEMpi7jbVt_RA,408
 upgini/utils/postal_code_utils.py,sha256=_8CR9tBqsPptQsmMUvnrCAmBaMIQSWH3JfJ4ly3x_zs,409
 upgini/utils/target_utils.py,sha256=3eHrDy_Dc9ozuOwHGnGA705m9glCxKmjB-DfLrflqiA,1370
-upgini/utils/track_info.py,sha256=dfDwLdQX7aY0eRT1izsMpeEiN0bYx_2YH7O69AoVhlU,4465
+upgini/utils/track_info.py,sha256=O_oL4gy1jH0DVgtiUeZAW0YKCeRT4B_bzH_SZYkFaOE,4076
 upgini/utils/warning_counter.py,sha256=vnmdFo5-7GBkU2bK9h_uC0K0Y_wtfcYstxOdeRfacO0,228
-upgini-1.1.132.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.132.dist-info/METADATA,sha256=sUIjt9bSFBtawDxGqez0g6UUJmfFXMHgnQd4W8x08IY,43227
-upgini-1.1.132.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-upgini-1.1.132.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
-upgini-1.1.132.dist-info/RECORD,,
+upgini-1.1.132a1.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.132a1.dist-info/METADATA,sha256=wEsepCklPcYgAQM8PGSh9OB2Fu6I8qmKNYqQKl867Do,43229
+upgini-1.1.132a1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+upgini-1.1.132a1.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
+upgini-1.1.132a1.dist-info/RECORD,,