upgini 1.2.145__tar.gz → 1.2.145a4065.dev1__tar.gz

{upgini-1.2.145 → upgini-1.2.145a4065.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.145
+Version: 1.2.145a4065.dev1
 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.2.145a4065.dev1/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.145a4065.dev1"

{upgini-1.2.145 → upgini-1.2.145a4065.dev1}/src/upgini/features_enricher.py

@@ -3000,16 +3000,15 @@ if response.status_code == 200:
 
         return selected_input_columns + selected_generated_features
 
-    def _validate_empty_search_keys(self, search_keys: dict[str, SearchKey], is_transform: bool = False):
+    def __validate_search_keys(self, search_keys: dict[str, SearchKey], search_id: str | None = None):
         if (search_keys is None or len(search_keys) == 0) and self.country_code is None:
-            if is_transform:
-                self.logger.debug("Transform started without search_keys")
+            if search_id:
+                self.logger.debug(f"search_id {search_id} provided without search_keys")
                 return
-            else:
-                self.logger.warning("search_keys not provided")
-                raise ValidationError(self.bundle.get("empty_search_keys"))
+            # else:
+            #     self.logger.warning("search_keys not provided")
+            #     raise ValidationError(self.bundle.get("empty_search_keys"))
 
-    def __validate_search_keys(self, search_keys: dict[str, SearchKey], search_id: str | None = None):
         key_types = search_keys.values()
 
         # Multiple search keys allowed only for PHONE, IP, POSTAL_CODE, EMAIL, HEM
@@ -4802,8 +4801,6 @@ if response.status_code == 200:
 
         self.logger.info(f"Prepared search keys: {valid_search_keys}")
 
-        self._validate_empty_search_keys(valid_search_keys, is_transform=is_transform)
-
         return valid_search_keys
 
     def __show_metrics(
@@ -4944,6 +4941,10 @@ if response.status_code == 200:
                 )
                 self.__log_warning(self.bundle.get("phone_detected_not_registered"))
 
+        if (search_keys is None or len(search_keys) == 0) and self.country_code is None:
+            self.logger.warning("search_keys not provided")
+            raise ValidationError(self.bundle.get("empty_search_keys"))
+
         return search_keys
 
     def _validate_binary_observations(self, y, task_type: ModelTaskType):

upgini-1.2.145/src/upgini/__about__.py

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