upgini 1.1.253a4__tar.gz → 1.1.253a3233__tar.gz

{upgini-1.1.253a4/src/upgini.egg-info → upgini-1.1.253a3233}/PKG-INFO

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

{upgini-1.1.253a4 → upgini-1.1.253a3233}/setup.py

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

{upgini-1.1.253a4 → upgini-1.1.253a3233}/src/upgini/utils/target_utils.py

@@ -152,7 +152,7 @@ def balance_undersample(
         # fill up to min_sample_threshold by majority class
         minority_class = df[df[target_column] == min_class_value]
         majority_class = df[df[target_column] != min_class_value]
-        sample_size = min(len(majority_class, min_sample_threshold - min_class_count))
+        sample_size = min(len(majority_class), min_sample_threshold - min_class_count)
         sampled_majority_class = majority_class.sample(
             n=sample_size, random_state=random_state
         )

{upgini-1.1.253a4 → upgini-1.1.253a3233/src/upgini.egg-info}/PKG-INFO

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

{upgini-1.1.253a4 → upgini-1.1.253a3233}/tests/test_features_enricher.py

@@ -2136,6 +2136,8 @@ def test_idempotent_order_with_imbalanced_dataset(requests_mock: Mocker):
         expected_result_df = (
             pd.read_parquet(expected_result_path).sort_values(by="system_record_id").reset_index(drop=True)
         )
+        expected_result_df["phone_num_a54a33"] = expected_result_df["phone_num_a54a33"].astype("Int64")
+        expected_result_df["rep_date_f5d6bb"] = expected_result_df["rep_date_f5d6bb"].astype("Int64")
 
         def test(n_shuffles: int):
             train_df = initial_train_df.copy()
@@ -2574,9 +2576,9 @@ def test_unsupported_arguments(requests_mock: Mocker):
 
     df = pd.DataFrame(
         {
-            "date": ["2021-01-01", "2021-01-02", "2023-01-01", "2023-01-02"],
-            "feature": [11, 10, 12, 13],
-            "target": [0, 1, 0, 1],
+            "date": ["2021-01-01", "2021-01-02", "2023-01-01", "2023-01-02", "2023-01-03"],
+            "feature": [11, 10, 12, 13, 14],
+            "target": [0, 1, 0, 1, 0],
         }
     )
 
@@ -2603,13 +2605,16 @@ def test_unsupported_arguments(requests_mock: Mocker):
 
         enricher.transform(df.drop(columns="target"), "unsupported_positional_argument", unsupported_key_argument=False)
 
-        enricher.calculate_metrics(
-            df.drop(columns="target"),
-            df["target"],
-            [(df.drop(columns="target"), df["target"])],
-            "unsupported_positional_argument",
-            unsupported_key_argument=False,
-        )
+        with pytest.raises(
+            ValueError, match="Only one class present in y_true. ROC AUC score is not defined in that case."
+        ):
+            enricher.calculate_metrics(
+                df.drop(columns="target"),
+                df["target"],
+                [(df.drop(columns="target"), df["target"])],
+                "unsupported_positional_argument",
+                unsupported_key_argument=False,
+            )
     finally:
         Dataset.MIN_ROWS_COUNT = original_min_rows
 

{upgini-1.1.253a4 → upgini-1.1.253a3233}/tests/test_metrics.py

@@ -258,9 +258,9 @@ def test_demo_metrics(requests_mock: Mocker):
             segment_header: [train_segment],
             rows_header: [464],
             target_mean_header: [100.7802],
-            baseline_mae: [21.5089],
-            enriched_mae: [20.8841],
-            uplift: [0.624774],
+            baseline_mae: [21.18862],
+            enriched_mae: [20.862484],
+            uplift: [0.326136],
         }
     )