upgini 1.1.292__py3-none-any.whl → 1.1.294__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.292"
+__version__ = "1.1.294"

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

@@ -95,6 +95,11 @@ class DataSourcePublisher:
                         "With MSISDN and DATE keys one of the snapshot_frequency_days or"
                         " join_date_abs_limit_days parameters is required"
                     )
+                if (
+                    set(search_keys.values()) == {SearchKey.PHONE, SearchKey.DATE}
+                    or set(search_keys.values()) == {SearchKey.HEM, SearchKey.DATE}
+                ) and not date_format:
+                    raise ValidationError("date_format argument is required for PHONE+DATE and HEM+DATE search keys")
 
                 request = {
                     "dataTableUri": data_table_uri,

upgini/features_enricher.py

@@ -2557,9 +2557,6 @@ class FeaturesEnricher(TransformerMixin):
         return [c for c, v in search_keys_with_autodetection.items() if v.value.value in keys]
 
     def _validate_X(self, X, is_transform=False) -> pd.DataFrame:
-        if _num_samples(X) == 0:
-            raise ValidationError(self.bundle.get("x_is_empty"))
-
         if isinstance(X, pd.DataFrame):
             if isinstance(X.columns, pd.MultiIndex) or isinstance(X.index, pd.MultiIndex):
                 raise ValidationError(self.bundle.get("x_multiindex_unsupported"))
@@ -2573,6 +2570,9 @@ class FeaturesEnricher(TransformerMixin):
         else:
             raise ValidationError(self.bundle.get("unsupported_x_type").format(type(X)))
 
+        if _num_samples(X) == 0:
+            raise ValidationError(self.bundle.get("x_is_empty"))
+
         if len(set(validated_X.columns)) != len(validated_X.columns):
             raise ValidationError(self.bundle.get("x_contains_dup_columns"))
         if not is_transform and not validated_X.index.is_unique:

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.292.dist-info → upgini-1.1.294.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.292
+Version: 1.1.294
 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.292.dist-info → upgini-1.1.294.dist-info}/RECORD

@@ -1,10 +1,11 @@
-upgini/__about__.py,sha256=ftI-AhPhhGVokbUGBIG7BOU1TcKcaEDEyviqLQdHimM,24
-upgini/__init__.py,sha256=asENHgEVHQBIkV-e_0IhE_ZWqkCG6398U3ZLrNzAH6k,407
+upgini/__about__.py,sha256=2vfGkEyxotARb9WDVVoCNdQiOrrhNqGA_vyl426_y5w,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/features_enricher.py,sha256=rLy6BwhL94VkRCp8W4RxJ0lBb7sZqTOsz-bUVI8nXZU,177530
 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=2BCFcglyZC1T5WYerqd_GGC6dg22o2Mh-ce53wOIOjQ,17465
+upgini/data_source/data_source_publisher.py,sha256=6paupnciqagACnSzjGSkA2a5i-c9ETvZheLqBYOJemk,17810
 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.292.dist-info/METADATA,sha256=VsA4dkEyCQEUAunLd-p24iOLBHnVFFMMVCb5SisZjck,48117
-upgini-1.1.292.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.1.292.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.292.dist-info/RECORD,,
+upgini-1.1.294.dist-info/METADATA,sha256=mOES85LZo9_PlVh9tpbFPRp2QI16iINyIk5ywPby7Dc,48117
+upgini-1.1.294.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.1.294.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.294.dist-info/RECORD,,

{upgini-1.1.292.dist-info → upgini-1.1.294.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.2
+Generator: hatchling 1.25.0
 Root-Is-Purelib: true
 Tag: py3-none-any