upgini 1.1.291a3232.post4__tar.gz → 1.1.293__tar.gz

{upgini-1.1.291a3232.post4 → upgini-1.1.293}/PKG-INFO

@@ -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.293/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.293"

upgini-1.1.293/src/upgini/__init__.py

@@ -0,0 +1,7 @@
+from .lazy_import import LazyImport
+
+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-1.1.291a3232.post4 → upgini-1.1.293}/src/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-1.1.291a3232.post4 → upgini-1.1.293}/src/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-1.1.293/src/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/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.291a3232-4"

upgini-1.1.291a3232.post4/src/upgini/__init__.py

@@ -1,21 +0,0 @@
-from typing import List
-
-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 []