upgini 1.1.291a3232.post4__py3-none-any.whl → 1.1.293__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.291a3232-4"
+__version__ = "1.1.293"

upgini/__init__.py

@@ -1,21 +1,7 @@
-from typing import List
+from .lazy_import import LazyImport
 
-from .dataset import Dataset
-from .features_enricher import FeaturesEnricher  # noqa: F401
-from .metadata import (  # noqa: F401
-    FileColumnMeaningType,
-    FileMetrics,
-    ModelTaskType,
-    SearchKey,
-)
-from .search_task import SearchTask
-
-
-def search_history() -> List[SearchTask]:
-    # TODO
-    return []
-
-
-def datasets_history() -> List[Dataset]:
-    # TODO
-    return []
+FeaturesEnricher = LazyImport('upgini.features_enricher', 'FeaturesEnricher')
+SearchKey = LazyImport('upgini.metadata', 'SearchKey')
+RuntimeParameters = LazyImport('upgini.metadata', 'RuntimeParameters')
+CVType = LazyImport('upgini.metadata', 'CVType')
+ModelTaskType = LazyImport('upgini.metadata', 'ModelTaskType')

upgini/data_source/data_source_publisher.py

@@ -347,3 +347,29 @@ class DataSourcePublisher:
             self.upload_online(search_keys=keys)
 
         print("All ADS-es successfully uploaded")
+
+    def union_search_tasks(
+        self,
+        search_ids: List[str],
+        target_user_email: str,
+        selected_features: Optional[List[str]] = None,
+        exclude_features: Optional[List[str]] = None,
+    ) -> str:
+        if not search_ids:
+            raise Exception("Empty search ids list")
+
+        if not target_user_email:
+            raise Exception("Empty target user email")
+
+        request = {
+            "search_task_ids": search_ids,
+            "target_user_email": target_user_email,
+        }
+        if selected_features:
+            request["selected_features"] = selected_features
+        if exclude_features:
+            request["exclude_features"] = exclude_features
+
+        response = self._rest_client.union_search_tasks(request, "trace_id")
+        print(response)
+        return response

upgini/http.py

@@ -291,6 +291,7 @@ class _RestClient:
     GET_ACTIVE_ADS_DEFINITIONS_URI = "private/api/v2/ads/definitions"
     UPLOAD_ONLINE_URI = "private/api/v2/ads/upload-online"
     STOP_ADS_MANAGEMENT_TASK_URI_FMT = "private/api/v2/ads/management-task/{0}/stop"
+    UNION_SEARCH_TASKS_URI_FMT = SERVICE_ROOT_V2 + "search/merge"
 
     ACCESS_TOKEN_HEADER_NAME = "Authorization"
     CONTENT_TYPE_HEADER_NAME = "Content-Type"
@@ -760,6 +761,10 @@ class _RestClient:
         api_path = self.STOP_ADS_MANAGEMENT_TASK_URI_FMT.format(ads_management_task_id)
         self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))
 
+    def union_search_tasks(self, request: dict, trace_id: str):
+        api_path = self.UNION_SEARCH_TASKS_URI_FMT
+        return self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id, request, result_format=None))
+
     # ---
 
     def _send_get_req(self, api_path: str, trace_id: Optional[str], additional_headers: Optional[dict] = None):

upgini/lazy_import.py

@@ -0,0 +1,22 @@
+import importlib
+
+
+class LazyImport:
+    def __init__(self, module_name, class_name):
+        self.module_name = module_name
+        self.class_name = class_name
+        self._module = None
+        self._class = None
+
+    def _load(self):
+        if self._module is None:
+            self._module = importlib.import_module(self.module_name)
+            self._class = getattr(self._module, self.class_name)
+
+    def __call__(self, *args, **kwargs):
+        self._load()
+        return self._class(*args, **kwargs)
+
+    def __getattr__(self, name):
+        self._load()
+        return getattr(self._class, name)

{upgini-1.1.291a3232.post4.dist-info → upgini-1.1.293.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.291a3232.post4
+Version: 1.1.293
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

{upgini-1.1.291a3232.post4.dist-info → upgini-1.1.293.dist-info}/RECORD

@@ -1,10 +1,11 @@
-upgini/__about__.py,sha256=jJbyMqOUOTTjUS0541LO2dVvKHQKVhOwnhdh0afozrE,31
-upgini/__init__.py,sha256=asENHgEVHQBIkV-e_0IhE_ZWqkCG6398U3ZLrNzAH6k,407
+upgini/__about__.py,sha256=qZMAUJW6Js0WNRcvGFbeSyGFbiVxGPjSbHngvQV4JrY,24
+upgini/__init__.py,sha256=ObEtjFkIssl83qeKNMLpIQygfwK8TzztwiI43YTsAP0,353
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=7TLVVhGtjgx_9yaiaIUK3kZSe_R9wg5dY0d4F5qCGM4,45636
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
 upgini/features_enricher.py,sha256=Bfqbzqj3h1Ox3s6hBA4UYXkG7hQDqGPB4JK1YXtxXxw,177530
-upgini/http.py,sha256=khrYSldpY-HbVLCcApfV1BjBFK6Uyuatb4colKybxgY,42301
+upgini/http.py,sha256=bp6jWl422Icy3AhHMdCcJv5NjExE45gSMmzMTPJjPuk,42600
+upgini/lazy_import.py,sha256=EwoM0msNGbSmWBhGbrLDny1DSnOlvTxCjmMKPxYlDms,610
 upgini/metadata.py,sha256=qDAIO7NLSSQp_XiXCv3U4XJTLO0KH3YuQ8lvCLYPqzs,9781
 upgini/metrics.py,sha256=DLvA2YLV4f7lnzBCcfZ5T4NkqAv3pbstbjTepavuT7U,30688
 upgini/search_task.py,sha256=LtRJ9bCPjMo1gJ-sUDKERhDwGcWKImrzwVFHjkMSQHQ,17071
@@ -22,7 +23,7 @@ upgini/autofe/operand.py,sha256=MKEsl3zxpWzRDpTkE0sNJxTu62U20sWOvEKhPjUWS6s,2915
 upgini/autofe/unary.py,sha256=ZWjLd-CUkNt_PpM8YuWLLipW1v_RdBlsl4JxXIVo9aM,3652
 upgini/autofe/vector.py,sha256=dLxfAstJs-gw_OQ1xxoxcM6pVzORlV0HVzdzt7cLXVQ,606
 upgini/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-upgini/data_source/data_source_publisher.py,sha256=B4fJ1owDCF5ZZ0Ca9ywi_CXVt4iPvABh5BGTnXdXmHk,16635
+upgini/data_source/data_source_publisher.py,sha256=2BCFcglyZC1T5WYerqd_GGC6dg22o2Mh-ce53wOIOjQ,17465
 upgini/mdc/__init__.py,sha256=aM08nIWFc2gWdWUa3_IuEnNND0cQPkBGnYpRMnfFN8k,1019
 upgini/mdc/context.py,sha256=3u1B-jXt7tXEvNcV3qmR9SDCseudnY7KYsLclBdwVLk,1405
 upgini/normalizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -56,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=Y96_PJ5cC-WsEbeqg20v9uqywDQobLoTb-xoP7S3o4E,7807
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.1.291a3232.post4.dist-info/METADATA,sha256=aKEzY66AyuXkd0_pSeuHIMhby3Czc7LM7iI5ui7qemg,48128
-upgini-1.1.291a3232.post4.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.1.291a3232.post4.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.291a3232.post4.dist-info/RECORD,,
+upgini-1.1.293.dist-info/METADATA,sha256=xxqpODSdTGvAAXew_U-IjsfNzYFALrNqOS2BK81cgy8,48117
+upgini-1.1.293.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+upgini-1.1.293.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.293.dist-info/RECORD,,