upgini 1.1.204__py3-none-any.whl → 1.1.205__py3-none-any.whl

upgini/http.py

@@ -886,15 +886,15 @@ class BackendLogHandler(logging.Handler):
         super().__init__(*args, **kwargs)
         self.rest_client = rest_client
         self.track_metrics = None
-        self.hostname = client_ip
+        self.hostname = "0.0.0.0"
+        self.client_ip = client_ip
 
     def emit(self, record: logging.LogRecord) -> None:
         def task():
             try:
                 if self.track_metrics is None or len(self.track_metrics) == 0:
-                    self.track_metrics = get_track_metrics()
-                    if self.hostname is None:
-                        self.hostname = self.track_metrics.get("ip") or "0.0.0.0"
+                    self.track_metrics = get_track_metrics(self.client_ip)
+                    self.hostname = self.track_metrics.get("ip") or "0.0.0.0"
                 text = self.format(record)
                 tags = self.track_metrics
                 tags["version"] = __version__

upgini/utils/display_utils.py

@@ -264,9 +264,6 @@ def prepare_and_show_report(
 def show_button_download_pdf(source: str, title="\U0001F4CA Download PDF report"):
     from IPython.display import HTML, display
 
-    with open("/Users/nikolaytoroptsev/Downloads/pdf.html", "w") as f:
-        f.write(source)
-
     file_name = f"upgini-report-{uuid.uuid4()}.pdf"
     with open(file_name, "wb") as output:
         pisa.CreatePDF(src=BytesIO(source.encode("UTF-8")), dest=output)

upgini/utils/track_info.py

@@ -5,6 +5,7 @@ import socket
 import sys
 from getpass import getuser
 from hashlib import sha256
+from typing import Optional
 from uuid import getnode
 
 from requests import get, post
@@ -46,7 +47,7 @@ def _get_execution_ide() -> str:
 
 
 @lru_cache()
-def get_track_metrics() -> dict:
+def get_track_metrics(client_ip: Optional[str]) -> dict:
     # default values
     track = {"ide": _get_execution_ide()}
     ident_res = "https://api.ipify.org"
@@ -93,28 +94,34 @@ def get_track_metrics() -> dict:
         except Exception as e:
             track["err"] = str(e)
             track["visitorId"] = "None"
-        try:
-            from google.colab import output  # type: ignore
-            from IPython.display import Javascript, display
-
-            display(
-                Javascript(
-                    f"""
-                        window.clientIP = fetch("{ident_res}")
-                            .then(response => response.text())
-                            .then(data => data);
-                    """
+        if client_ip:
+            track["ip"] = client_ip
+        else:
+            try:
+                from google.colab import output  # type: ignore
+                from IPython.display import Javascript, display
+
+                display(
+                    Javascript(
+                        f"""
+                            window.clientIP = fetch("{ident_res}")
+                                .then(response => response.text())
+                                .then(data => data);
+                        """
+                    )
                 )
-            )
-            track["ip"] = output.eval_js("window.clientIP", timeout_sec=10)
-        except Exception as e:
-            track["err"] = str(e)
-            track["ip"] = "0.0.0.0"
+                track["ip"] = output.eval_js("window.clientIP", timeout_sec=10)
+            except Exception as e:
+                track["err"] = str(e)
+                track["ip"] = "0.0.0.0"
 
     elif track["ide"] == "binder":
         try:
             if "CLIENT_IP" in os.environ.keys():
-                track["ip"] = os.environ["CLIENT_IP"]
+                if client_ip:
+                    track["ip"] = client_ip
+                else:
+                    track["ip"] = os.environ["CLIENT_IP"]
                 track["visitorId"] = sha256(os.environ["CLIENT_IP"].encode()).hexdigest()
         except Exception as e:
             track["err"] = str(e)
@@ -142,7 +149,10 @@ def get_track_metrics() -> dict:
             track["visitorId"] = "None"
     else:
         try:
-            track["ip"] = get(ident_res, timeout=10).text
+            if client_ip:
+                track["ip"] = client_ip
+            else:
+                track["ip"] = get(ident_res, timeout=10).text
             track["visitorId"] = sha256(str(getnode()).encode()).hexdigest()
         except Exception as e:
             track["err"] = str(e)

{upgini-1.1.204.dist-info → upgini-1.1.205.dist-info}/METADATA

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

{upgini-1.1.204.dist-info → upgini-1.1.205.dist-info}/RECORD

@@ -4,7 +4,7 @@ upgini/dataset.py,sha256=e5PX-NICeUUUrHJwEMpndYkNvWwbf-KE8USUd1rleSc,47502
 upgini/errors.py,sha256=BqpvfhW2jJW5fa5KXj0alhXatGl-WK4xTl309-QNLp8,959
 upgini/features_enricher.py,sha256=EyU1QEye4slsDAYUsXcJIFvIaY0dwIWoXxJ2DRAKXKE,146753
 upgini/fingerprint.js,sha256=VygVIQlN1v4NGZfjHqtRogOw8zjTnnMNJg_f7M5iGQU,33442
-upgini/http.py,sha256=58K9axk6mBX1lXW-aGejBzYBeFsgWhrZkCP7eWhBfT0,39701
+upgini/http.py,sha256=adz7jx0R7y3ndpXxt6aq2BxsRZea5H5Fkj4ETCqk8xQ,39700
 upgini/metadata.py,sha256=9Hm9L7efMGIRSfPUYZ_RUBW7_WMeeTL1fGhylMSvR7w,6135
 upgini/metrics.py,sha256=JeZz3hVAOBKOTtX60fn9jIHMHjt3LyIOfy_7VSpWbXI,22022
 upgini/search_task.py,sha256=f-wARUaAMKmSNQNmA_pmtVBhWpoSBY2F8S3DCt_y9zI,16799
@@ -32,7 +32,7 @@ upgini/utils/country_utils.py,sha256=9BXSXoGm3nVoOZE_bRENY-KMkwMUFvAF3Au0zxUNA1o
 upgini/utils/custom_loss_utils.py,sha256=DBslpjWGPt7xTeypt78baR59012SYphbPsO_YLKdilo,3972
 upgini/utils/cv_utils.py,sha256=X-u6Ndnv9jcQoE0jYS11zbyVd4d6IZ4UALn4raLqZ2I,2393
 upgini/utils/datetime_utils.py,sha256=wS_wBme83H3n3eAVDTL1QFjCG6oFLJ8bxB9Aqs9z-sg,5942
-upgini/utils/display_utils.py,sha256=6A--wHsmLfOTYw28TVGxtb-o-Ig_6DriRZ3VfLgXVCQ,9424
+upgini/utils/display_utils.py,sha256=KAuyrFy6BHpCNVk6upvpc0L2ukG1Ab7RQK7rxD9i37M,9328
 upgini/utils/email_utils.py,sha256=MhCLUAWqbp81xRyKizauNhVx6t_MFeJQRQ8pFM7EpFo,3480
 upgini/utils/fallback_progress_bar.py,sha256=f-VzVbiO6oU9WoKzEgoegYotixdiKanGlvdQCOGC-NY,1128
 upgini/utils/features_validator.py,sha256=iP8muF3PUf_aP9m7O3i3LPMuJPTNbw8rCAWqgvDt_h8,2369
@@ -43,10 +43,10 @@ upgini/utils/postal_code_utils.py,sha256=_8CR9tBqsPptQsmMUvnrCAmBaMIQSWH3JfJ4ly3
 upgini/utils/progress_bar.py,sha256=iNXyqT3vKCeHpfiG5HHwr7Lk2cTtKViM93Fl8iZnjGc,1564
 upgini/utils/sklearn_ext.py,sha256=IMx2La70AXAggApVpT7sMEjWqVWon5AMZt4MARDsIMQ,43847
 upgini/utils/target_utils.py,sha256=cu52icjhDIPpEStHYMXrD2hIl9gzvfnxZr0Ra5osV0k,1616
-upgini/utils/track_info.py,sha256=2IGGyHPXBLhWcLO8-Q-5qir52k_kD6DtdU-sv_Z2hHY,5325
+upgini/utils/track_info.py,sha256=87ZFhpoi1g0Z3kl34CwAN60ZelBUvfuQNzkuloZrQQ8,5701
 upgini/utils/warning_counter.py,sha256=vnmdFo5-7GBkU2bK9h_uC0K0Y_wtfcYstxOdeRfacO0,228
-upgini-1.1.204.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.204.dist-info/METADATA,sha256=Itf-VGrUj_QpZT2wgsttyVlDOvvbFuKXM2b7LgndFDs,48306
-upgini-1.1.204.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-upgini-1.1.204.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
-upgini-1.1.204.dist-info/RECORD,,
+upgini-1.1.205.dist-info/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.205.dist-info/METADATA,sha256=R3mdJs-XrPe9Q8bc_jCvKGXDkigCONQRJfSLNjZs_C8,48306
+upgini-1.1.205.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+upgini-1.1.205.dist-info/top_level.txt,sha256=OFhTGiDIWKl5gFI49qvWq1R9IKflPaE2PekcbDXDtx4,7
+upgini-1.1.205.dist-info/RECORD,,