upgini 1.2.130__tar.gz → 1.2.132__tar.gz

{upgini-1.2.130 → upgini-1.2.132}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.130
+Version: 1.2.132
 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.132/src/upgini/__about__.py

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

{upgini-1.2.130 → upgini-1.2.132}/src/upgini/features_enricher.py

@@ -732,6 +732,7 @@ class FeaturesEnricher(TransformerMixin):
 
             result = self.transform(
                 X,
+                y,
                 exclude_features_sources=exclude_features_sources,
                 keep_input=keep_input,
                 trace_id=trace_id,
@@ -2786,7 +2787,7 @@ if response.status_code == 200:
             df_without_features = df.drop(columns=features_not_to_pass, errors="ignore")
 
             df_without_features, full_duplicates_warning = clean_full_duplicates(
-                df_without_features, self.logger, bundle=self.bundle
+                df_without_features, is_transform=True, logger=self.logger, bundle=self.bundle
             )
             if not silent_mode and full_duplicates_warning:
                 self.__log_warning(full_duplicates_warning)

{upgini-1.2.130 → upgini-1.2.132}/src/upgini/utils/deduplicate_utils.py

@@ -160,7 +160,10 @@ def remove_fintech_duplicates(
 
 
 def clean_full_duplicates(
-    df: pd.DataFrame, logger: Optional[Logger] = None, bundle: Optional[ResourceBundle] = None
+    df: pd.DataFrame,
+    is_transform: bool = False,
+    logger: Optional[Logger] = None,
+    bundle: Optional[ResourceBundle] = None,
 ) -> Tuple[pd.DataFrame, Optional[str]]:
     if logger is None:
         logger = logging.getLogger()
@@ -193,7 +196,7 @@ def clean_full_duplicates(
         logger.warning(bundle.get("dataset_full_duplicates").format(share_full_dedup))
 
     msg = None
-    if TARGET in df.columns:
+    if not is_transform and TARGET in df.columns:
         unique_columns.remove(TARGET)
 
         # Separate rows to exclude from deduplication:

upgini-1.2.130/src/upgini/__about__.py

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