upgini 1.2.122a1__py3-none-any.whl → 1.2.122a2__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.122a1"
+__version__ = "1.2.122a2"

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/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.dist-info → upgini-1.2.122a2.dist-info}/METADATA

@@ -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.122a1.dist-info → upgini-1.2.122a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=hzzmPAt8OZIX5YRwSKl5dj9LWowWDEnOpFN5Xq2xARQ,26
+upgini/__about__.py,sha256=s0uuov4eE8uySaKt-V4Lx3FlZDRMujpgzcGDSSewW1w,26
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=pQ8JQe0cdygD-W9GefJmfE6bnj4EYzXsjlgWdIS9nS8,31578
@@ -16,9 +16,9 @@ upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operators.py,sha256=rdjF5eaE4bC6Q4eu_el5Z7ekYt8DjOFermz2bePPbUc,333
 upgini/autofe/binary.py,sha256=oOEECc4nRzZN2tYaiqx8F2XHnfWpk1bVvb7ZkZJ0lO8,7709
 upgini/autofe/date.py,sha256=RvexgrL1_6ISYPVrl9HUQmPgpVSGQsTNv8YhNQWs-5M,11329
-upgini/autofe/feature.py,sha256=2jOdTTnUqdUewznxsveuTLgKcPLPNtFWS0YQsYYBbPk,16622
+upgini/autofe/feature.py,sha256=32uZfa7isXcMpvZiQyMYD3M02lIWjEc0vAheBr07ZP8,16662
 upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
-upgini/autofe/operator.py,sha256=RB3rKMjFi5Cx81RiYXN3OTCuXjmvzmFKQrxn4h0Oclo,5219
+upgini/autofe/operator.py,sha256=R2VoE3QAUoHPzzcmzRWROGq2OmegZOdZlTLRDZfrM7s,6279
 upgini/autofe/unary.py,sha256=FFtvkQaT0cu_zPZ1jCLcsjik-UUh12qQFF3tUW8NqsE,6675
 upgini/autofe/utils.py,sha256=dYrtyAM8Vcc_R8u4dNo54IsGrHKagTHDJTKhGho0bRg,2967
 upgini/autofe/vector.py,sha256=r5H6DKT5f3KNjERpV2OOloZ96nDWkModXnpsqw_A77Q,2313
@@ -74,7 +74,7 @@ upgini/utils/target_utils.py,sha256=GCPn4QeJ83JJ_vyBJ3IhY5fyIRkLC9q9BE59S2FRO1I,
 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.122a1.dist-info/METADATA,sha256=3pPdEVaYucgJB5Klks339i5-JTM7hJpEZUmZS7dEWi8,50745
-upgini-1.2.122a1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.122a1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.122a1.dist-info/RECORD,,
+upgini-1.2.122a2.dist-info/METADATA,sha256=Nd9Mzf8ebdrmu35KxUkg72unHIM5eVAuoOnpw6gKW3o,50745
+upgini-1.2.122a2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.122a2.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.122a2.dist-info/RECORD,,