upgini 1.2.9a2__tar.gz → 1.2.9a3__tar.gz

{upgini-1.2.9a2 → upgini-1.2.9a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.9a2
+Version: 1.2.9a3
 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.9a3/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.9a3"

{upgini-1.2.9a2 → upgini-1.2.9a3}/src/upgini/__init__.py

@@ -1,7 +1,7 @@
 import os
 
-# from upgini.features_enricher import FeaturesEnricher  # noqa: F401
-# from upgini.metadata import SearchKey, CVType, RuntimeParameters, ModelTaskType  # noqa: F401
+from upgini.features_enricher import FeaturesEnricher  # noqa: F401
+from upgini.metadata import SearchKey, CVType, RuntimeParameters, ModelTaskType  # noqa: F401
 # from .lazy_import import LazyImport
 
 os.environ["SETUPTOOLS_USE_DISTUTILS"] = "stdlib"

{upgini-1.2.9a2 → upgini-1.2.9a3}/src/upgini/features_enricher.py

@@ -2531,7 +2531,7 @@ class FeaturesEnricher(TransformerMixin):
         features_columns = [c for c in df.columns if c not in non_feature_columns]
 
         features_to_drop = FeaturesValidator(self.logger).validate(
-            df, features_columns, self.generate_features, self.warning_counter
+            df, features_columns, self.generate_features, columns_renaming, self.warning_counter
         )
         self.fit_dropped_features.update(features_to_drop)
         df = df.drop(columns=features_to_drop)
@@ -2657,7 +2657,7 @@ class FeaturesEnricher(TransformerMixin):
             and len(self._search_task.unused_features_for_generation) > 0
         ):
             unused_features_for_generation = [
-                dataset.columns_renaming.get(col) or col for col in self._search_task.unused_features_for_generation
+                columns_renaming.get(col) or col for col in self._search_task.unused_features_for_generation
             ]
             msg = self.bundle.get("features_not_generated").format(unused_features_for_generation)
             self.logger.warning(msg)

{upgini-1.2.9a2 → upgini-1.2.9a3}/src/upgini/utils/features_validator.py

@@ -1,6 +1,6 @@
 import logging
 from logging import Logger
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 import pandas as pd
 from pandas.api.types import is_integer_dtype, is_object_dtype, is_string_dtype
@@ -22,6 +22,7 @@ class FeaturesValidator:
         df: pd.DataFrame,
         features: List[str],
         features_for_generate: Optional[List[str]],
+        columns_renaming: Dict[str, str],
         warning_counter: WarningCounter,
     ) -> List[str]:
         # one_hot_encoded_features = []
@@ -63,7 +64,7 @@ class FeaturesValidator:
 
         high_cardinality_features = self.find_high_cardinality(df[features])
         if features_for_generate:
-            high_cardinality_features = [f for f in high_cardinality_features if f not in features_for_generate]
+            high_cardinality_features = [f for f in high_cardinality_features if columns_renaming[f] not in features_for_generate]
         if high_cardinality_features:
             msg = bundle.get("high_cardinality_features").format(high_cardinality_features)
             print(msg)

upgini-1.2.9a2/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.9a2"