upgini 1.1.300__py3-none-any.whl → 1.1.302__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.300"
+__version__ = "1.1.302"

upgini/features_enricher.py

@@ -3186,7 +3186,9 @@ class FeaturesEnricher(TransformerMixin):
 
         if rows_to_drop is not None:
             self.logger.info(f"Before dropping target outliers size: {len(result_features)}")
-            result_features = result_features[~result_features[SYSTEM_RECORD_ID].isin(rows_to_drop[SYSTEM_RECORD_ID])]
+            result_features = result_features[
+                ~result_features[ENTITY_SYSTEM_RECORD_ID].isin(rows_to_drop[ENTITY_SYSTEM_RECORD_ID])
+            ]
             self.logger.info(f"After dropping target outliers size: {len(result_features)}")
 
         result_eval_sets = {}
@@ -3219,11 +3221,11 @@ class FeaturesEnricher(TransformerMixin):
             result_train = result_train_features
 
         if drop_system_record_id:
-            if SYSTEM_RECORD_ID in result_train.columns:
-                result_train = result_train.drop(columns=SYSTEM_RECORD_ID)
+            result_train = result_train.drop(columns=[SYSTEM_RECORD_ID, ENTITY_SYSTEM_RECORD_ID], errors="ignore")
             for eval_set_index in result_eval_sets.keys():
-                if SYSTEM_RECORD_ID in result_eval_sets[eval_set_index].columns:
-                    result_eval_sets[eval_set_index] = result_eval_sets[eval_set_index].drop(columns=SYSTEM_RECORD_ID)
+                result_eval_sets[eval_set_index] = result_eval_sets[eval_set_index].drop(
+                    columns=[SYSTEM_RECORD_ID, ENTITY_SYSTEM_RECORD_ID], errors="ignore"
+                )
 
         return result_train, result_eval_sets
 
@@ -3481,6 +3483,9 @@ class FeaturesEnricher(TransformerMixin):
         is_transform=False,
         silent_mode=False,
     ):
+        for _, key_type in search_keys.items():
+            if not isinstance(key_type, SearchKey):
+                raise ValidationError(self.bundle.get("unsupported_type_of_search_key").format(key_type))
         valid_search_keys = {}
         unsupported_search_keys = {
             SearchKey.IP_RANGE_FROM,

upgini/resource_bundle/strings.properties

@@ -85,6 +85,7 @@ unregistered_only_personal_keys=Only personal search keys used. Api_key from pro
 search_key_not_found=Column `{}` from search_keys was not found in X dataframe: {}
 numeric_search_key_not_found=Index {} in search_keys is out of bounds for {} columns of X dataframe
 unsupported_search_key_type=Unsupported type of key in search_keys: {}
+unsupported_type_of_search_key=Unsupported type of search key: {}. It should be a member of SearcKey
 search_key_country_and_country_code=\nWARNING: SearchKey.COUNTRY and country_code parameter were passed simultaniously. Parameter country_code will be ignored
 empty_search_key=Search key {} is empty. Please fill values or remove this search key
 single_constant_search_key=\nWARNING: Constant value detected for the {} search key in the X dataframe: {}.\nThat search key will add constant features for different y values.\nPlease add extra search keys with non constant values, like the COUNTRY, POSTAL_CODE, DATE, PHONE NUMBER, EMAIL/HEM or IPv4

{upgini-1.1.300.dist-info → upgini-1.1.302.dist-info}/METADATA

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

@@ -1,9 +1,9 @@
-upgini/__about__.py,sha256=IcfjUPTVJrTs-NV6tcVvLWMSPbW14GYerq2VeSrrzc0,24
+upgini/__about__.py,sha256=-qFeTOSW0hTZsmYrk502kQwbd51FNaHb_oq4nYfeY38,24
 upgini/__init__.py,sha256=ObEtjFkIssl83qeKNMLpIQygfwK8TzztwiI43YTsAP0,353
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=MOzBVsvzlHLxNfPWtMaXC_jIPeW7_gUvbSGeXnsPgNI,46158
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
-upgini/features_enricher.py,sha256=nweJzEV8ZbDN5wvf5Gdf-HgcNz7711gNgmRxz4ZUopI,183112
+upgini/features_enricher.py,sha256=PRiQGAmnm9jPXN4Uqus5l8RQIPAWPxsIBnIoEMnh6g8,183342
 upgini/http.py,sha256=bp6jWl422Icy3AhHMdCcJv5NjExE45gSMmzMTPJjPuk,42600
 upgini/lazy_import.py,sha256=EwoM0msNGbSmWBhGbrLDny1DSnOlvTxCjmMKPxYlDms,610
 upgini/metadata.py,sha256=wOFCJruDBhC4Hiiiqf8GeHZnnm6rhJy8t6fg5B0Z4TQ,10209
@@ -30,7 +30,7 @@ upgini/normalizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 upgini/normalizer/phone_normalizer.py,sha256=EzTaahk6myRv6ZXgbyVFGY4kpo_2VlQgOrm5_lfbmNI,9996
 upgini/resource_bundle/__init__.py,sha256=S5F2G47pnJd2LDpmFsjDqEwiKkP8Hm-hcseDbMka6Ko,8345
 upgini/resource_bundle/exceptions.py,sha256=5fRvx0_vWdE1-7HcSgF0tckB4A9AKyf5RiinZkInTsI,621
-upgini/resource_bundle/strings.properties,sha256=6jYqcxj06ZopXwr5YYMGXX1QiNNJNFo2SuwAR0qleRk,26358
+upgini/resource_bundle/strings.properties,sha256=WZAuYPX2Dpn6BHoA3RX8uvMNMr-yJE2fF7Gz0i24x2s,26459
 upgini/resource_bundle/strings_widget.properties,sha256=gOdqvZWntP2LCza_tyVk1_yRYcG4c04K9sQOAVhF_gw,1577
 upgini/sampler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/sampler/base.py,sha256=7GpjYqjOp58vYcJLiX__1R5wjUlyQbxvHJ2klFnup_M,6389
@@ -57,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.300.dist-info/METADATA,sha256=hLE9o5ZxxN1PnP4WR6ZUnp5yFpEb1cbVSwWEOiQDZBE,48153
-upgini-1.1.300.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.1.300.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.300.dist-info/RECORD,,
+upgini-1.1.302.dist-info/METADATA,sha256=Yf0pExfLbj0GqemC-7A1cIfyX1ZKQi_MmA7NNL0FM_g,48153
+upgini-1.1.302.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.1.302.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.302.dist-info/RECORD,,