upgini 1.2.122a1__tar.gz → 1.2.122a2__tar.gz

{upgini-1.2.122a1 → upgini-1.2.122a2}/PKG-INFO

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

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

{upgini-1.2.122a1 → upgini-1.2.122a2}/src/upgini/autofe/feature.py

@@ -145,6 +145,7 @@ class Feature:
         )
 
     def rename_columns(self, mapping: Dict[str, str]) -> "Feature":
+        self.op.rename_columns(mapping)
         for child in self.children:
             child.rename_columns(mapping)
         self.cached_display_name = None

{upgini-1.2.122a1 → upgini-1.2.122a2}/src/upgini/autofe/operator.py

@@ -89,6 +89,32 @@ class Operator(BaseModel, metaclass=OperatorRegistry):
     def delete_data(self):
         pass
 
+    def rename_columns(self, columns_renaming: Dict[str, str]) -> "Operator":
+        # Rename occurrences of column names inside self.params keys according to columns_renaming
+        if not self.params or not columns_renaming:
+            return self
+
+        # Replace longer keys first to avoid partial overlaps
+        replacements = sorted(columns_renaming.items(), key=lambda kv: -len(kv[0]))
+
+        renamed_params: Dict[str, str] = {}
+        for param_key, param_value in self.params.items():
+            new_key = param_key
+            for old, new in replacements:
+                if old and old in new_key:
+                    new_key = new_key.replace(old, new)
+
+            if new_key in renamed_params and new_key != param_key:
+                self._logger.warning(
+                    "Param key collision after rename: '%s' -> '%s'. Overwriting value.",
+                    param_key,
+                    new_key,
+                )
+            renamed_params[new_key] = param_value
+
+        self.params = renamed_params
+        return self
+
 
 class ParametrizedOperator(Operator, abc.ABC):
 

upgini-1.2.122a1/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.122a1"