upgini 1.1.233a1__tar.gz → 1.1.234a1__tar.gz

{upgini-1.1.233a1/src/upgini.egg-info → upgini-1.1.234a1}/PKG-INFO

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

{upgini-1.1.233a1 → upgini-1.1.234a1}/setup.py

@@ -40,7 +40,7 @@ def send_log(msg: str):
 
 
 here = Path(__file__).parent.resolve()
-version = "1.1.233a1"
+version = "1.1.234a1"
 try:
     send_log(f"Start setup PyLib version {version}")
     setup(

{upgini-1.1.233a1 → upgini-1.1.234a1}/src/upgini/features_enricher.py

@@ -25,7 +25,7 @@ from upgini.autofe.feature import Feature
 
 from upgini.data_source.data_source_publisher import CommercialSchema
 from upgini.dataset import Dataset
-from upgini.errors import ValidationError
+from upgini.errors import HttpError, ValidationError
 from upgini.http import (
     UPGINI_API_KEY,
     LoggerFactory,
@@ -245,6 +245,9 @@ class FeaturesEnricher(TransformerMixin):
                     # TODO validate search_keys with search_keys from file_metadata
                     print(bundle.get("search_by_task_id_finish"))
                     self.logger.info(f"Successfully initialized with search_id: {search_id}")
+                except HttpError as e:
+                    if "Interrupted by client" in e.args[0]:
+                        raise ValidationError("Search was cancelled")
                 except Exception as e:
                     print(bundle.get("failed_search_by_task_id"))
                     self.logger.exception(f"Failed to find search_id: {search_id}")
@@ -382,6 +385,8 @@ class FeaturesEnricher(TransformerMixin):
 
             self.logger.info("Start fit")
 
+            self.__validate_search_keys(self.search_keys, self.search_id)
+
             try:
                 self.X = X
                 self.y = y
@@ -511,6 +516,8 @@ class FeaturesEnricher(TransformerMixin):
 
             self.logger.info("Start fit_transform")
 
+            self.__validate_search_keys(self.search_keys, self.search_id)
+
             search_progress = SearchProgress(0.0, ProgressStage.START_FIT)
             if progress_callback is not None:
                 progress_callback(search_progress)
@@ -655,6 +662,8 @@ class FeaturesEnricher(TransformerMixin):
                 print(msg)
 
             self.logger.info("Start transform")
+
+            self.__validate_search_keys(self.search_keys, self.search_id)
             try:
                 if len(self.feature_names_) == 0:
                     self.logger.warning(bundle.get("no_important_features_for_transform"))
@@ -815,6 +824,8 @@ class FeaturesEnricher(TransformerMixin):
                 self.logger.warning(msg)
                 print(msg)
 
+            self.__validate_search_keys(self.search_keys, self.search_id)
+
             try:
                 self.__log_debug_information(
                     X if X is not None else self.X,
@@ -1914,7 +1925,7 @@ class FeaturesEnricher(TransformerMixin):
         if (search_keys is None or len(search_keys) == 0) and self.country_code is None:
             if search_id:
                 self.logger.warning(f"search_id {search_id} provided without search_keys")
-                raise ValidationError(bundle.get("search_key_differ_from_fit"))
+                return
             else:
                 self.logger.warning("search_keys not provided")
                 raise ValidationError(bundle.get("empty_search_keys"))

{upgini-1.1.233a1 → upgini-1.1.234a1/src/upgini.egg-info}/PKG-INFO

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