upgini 1.2.103a6__py3-none-any.whl → 1.2.105__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.103a6"
+__version__ = "1.2.105"

upgini/data_source/data_source_publisher.py

@@ -68,6 +68,7 @@ class DataSourcePublisher:
         date_features_format: Optional[str] = None,
         generate_runtime_embeddings: Optional[List[str]] = None,
         exclude_raw: Optional[List[str]] = None,
+        force_percentile_generation: Optional[List[str]] = None,
         _force_generation=False,
         _silent=False,
     ) -> str:
@@ -196,6 +197,8 @@ class DataSourcePublisher:
                     request["generateRuntimeEmbeddingsFeatures"] = generate_runtime_embeddings
                 if exclude_raw is not None:
                     request["excludeRaw"] = exclude_raw
+                if force_percentile_generation is not None:
+                    request["forcePercentileGeneration"] = force_percentile_generation
                 self.logger.info(f"Start registering data table {request}")
 
                 task_id = self._rest_client.register_ads(request, trace_id)
@@ -229,19 +232,28 @@ class DataSourcePublisher:
 
                         def on_button_clicked(b):
                             self.place(
-                                data_table_uri,
-                                search_keys,
-                                secondary_search_keys,
-                                sort_column,
-                                date_format,
-                                exclude_columns,
-                                hash_feature_names,
-                                snapshot_frequency_days,
-                                features_for_embeddings,
-                                data_table_id_to_replace,
-                                exclude_from_autofe_generation,
-                                True,
-                                _silent,
+                                data_table_uri=data_table_uri,
+                                search_keys=search_keys,
+                                update_frequency=update_frequency,
+                                exclude_from_autofe_generation=exclude_from_autofe_generation,
+                                secondary_search_keys=secondary_search_keys,
+                                sort_column=sort_column,
+                                date_format=date_format,
+                                exclude_columns=exclude_columns,
+                                hash_feature_names=hash_feature_names,
+                                snapshot_frequency_days=snapshot_frequency_days,
+                                join_date_abs_limit_days=join_date_abs_limit_days,
+                                features_for_embeddings=features_for_embeddings,
+                                data_table_id_to_replace=data_table_id_to_replace,
+                                keep_features=keep_features,
+                                date_features=date_features,
+                                date_vector_features=date_vector_features,
+                                date_features_format=date_features_format,
+                                generate_runtime_embeddings=generate_runtime_embeddings,
+                                exclude_raw=exclude_raw,
+                                force_percentile_generation=force_percentile_generation,
+                                _force_generation=True,
+                                _silent=_silent,
                             )
 
                         button.on_click(on_button_clicked)
@@ -255,7 +267,8 @@ class DataSourcePublisher:
                 print(msg)
                 if "warnings" in status_response and status_response["warnings"]:
                     self.logger.warning(status_response["warnings"])
-                    print(status_response["warnings"])
+                    for warning in status_response["warnings"]:
+                        print(warning)
                 return data_table_id
             except KeyboardInterrupt:
                 if task_id is not None:
@@ -501,6 +514,7 @@ class DataSourcePublisher:
         input_names: List[str],
         search_id: str,
         date_column: Optional[str] = None,
+        score_name: Optional[str] = None,
         model_type: Optional[Literal["ONNX", "CATBOOST"]] = None,
         description: str = "",
     ):
@@ -510,6 +524,7 @@ class DataSourcePublisher:
             "modelName": name,
             "inputNames": input_names,
             "dateColumn": date_column,
+            "scoreName": score_name,
             "searchTaskId": search_id,
             "modelType": model_type or "ONNX",
             "description": description,

upgini/features_enricher.py

@@ -2837,6 +2837,8 @@ if response.status_code == 200:
 
         validate_scoring_argument(scoring)
 
+        self._validate_baseline_score(validated_X, validated_eval_set)
+
         self.__log_debug_information(
             validated_X,
             validated_y,

upgini/metrics.py

@@ -475,6 +475,7 @@ class EstimatorWrapper:
         if baseline_score_column is not None and self.metric_name == "GINI":
             self.logger.info("Calculate baseline GINI on passed baseline_score_column and target")
             metric = roc_auc_score(y, x[baseline_score_column])
+            metric = self.post_process_metric(metric)
             metric_std = None
             average_shap_values = None
         else:

{upgini-1.2.103a6.dist-info → upgini-1.2.105.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.103a6
+Version: 1.2.105
 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.103a6.dist-info → upgini-1.2.105.dist-info}/RECORD

@@ -1,12 +1,12 @@
-upgini/__about__.py,sha256=YINSZx_dtM8hJaZdUvfPJhPcgsRxL7sluGJryc6JviA,26
+upgini/__about__.py,sha256=gxxeFXQgBe3cRhCPCpyOIotj3ufsQIw5lbHWL9O2HCo,24
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=e6JDYTZ2AwC5aF-dqclKZKkiKrHo2f6cFmMQO2ZZmjM,32724
 upgini/errors.py,sha256=2b_Wbo0OYhLUbrZqdLIx5jBnAsiD1Mcenh-VjR4HCTw,950
-upgini/features_enricher.py,sha256=qcS6QzfAZL60bXoupXRQVWyeOodFV-hgyajvsDHIGG8,220298
+upgini/features_enricher.py,sha256=dBCBqAhzVHqRB2b1sPy9wzSi9XtIzeb6uArmJhcjj_8,220370
 upgini/http.py,sha256=DNcoS7qdxG0mOJn6I8r6O5I6XdIJTdzDzW3hkz3NgG4,45443
 upgini/metadata.py,sha256=vsbbHyPCP3Rs8WkeDgQg99uAA_zmsbDStAT-NwDYhO4,12455
-upgini/metrics.py,sha256=gXr2aiw5j9QBWBo1hZp40Is679hef5q8MrT6LJfjsBk,45661
+upgini/metrics.py,sha256=V2SP6NS5bfFHzRqufeKVsCXME1yG4t_8Dmk2E3zKdYk,45715
 upgini/search_task.py,sha256=Q5HjBpLIB3OCxAD1zNv5yQ3ZNJx696WCK_-H35_y7Rs,17912
 upgini/spinner.py,sha256=4iMd-eIe_BnkqFEMIliULTbj6rNI2HkN_VJ4qYe0cUc,1118
 upgini/version_validator.py,sha256=DvbaAvuYFoJqYt0fitpsk6Xcv-H1BYDJYHUMxaKSH_Y,1509
@@ -31,7 +31,7 @@ upgini/autofe/timeseries/roll.py,sha256=zADKXU-eYWQnQ5R3am1yEal8uU6Tm0jLAixwPb_a
 upgini/autofe/timeseries/trend.py,sha256=K1_iw2ko_LIUU8YCUgrvN3n0MkHtsi7-63-8x9er1k4,2129
 upgini/autofe/timeseries/volatility.py,sha256=SvZfhM_ZAWCNpTf87WjSnZsnlblARgruDlu4By4Zvhc,8078
 upgini/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-upgini/data_source/data_source_publisher.py,sha256=YEnjqGW706Od6uTRp9l_TsTuppwSl5jRPZtg50Rngwk,25284
+upgini/data_source/data_source_publisher.py,sha256=suRmAF1i7yiZ8vJjpEKdVr5Wqtr7o1_vjAhaN9B4DU0,26518
 upgini/mdc/__init__.py,sha256=iHJlXQg6xRM1-ZOUtaPSJqw5SpQDszvxp4LyqviNLIQ,1027
 upgini/mdc/context.py,sha256=3u1B-jXt7tXEvNcV3qmR9SDCseudnY7KYsLclBdwVLk,1405
 upgini/normalizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -71,7 +71,7 @@ upgini/utils/target_utils.py,sha256=i3Xt5l9ybB2_nF_ma5cfPuL3OeFTs2dY2xDI0p4Azpg,
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
 upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
-upgini-1.2.103a6.dist-info/METADATA,sha256=HXXbQMsui50iMdVHRGQKUs7xoi_rXVQFFYwLxsG1YAQ,49531
-upgini-1.2.103a6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.103a6.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.103a6.dist-info/RECORD,,
+upgini-1.2.105.dist-info/METADATA,sha256=SaEsWWVsSk0fCFud9PY5M-zkFTFiIpHYaPCViAk6jkw,49529
+upgini-1.2.105.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.105.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.105.dist-info/RECORD,,