upgini 1.1.282a3418.post1__tar.gz → 1.1.282a3418.post2__tar.gz

{upgini-1.1.282a3418.post1 → upgini-1.1.282a3418.post2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.282a3418.post1
+Version: 1.1.282a3418.post2
 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.1.282a3418.post2/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.282a3418-2"

{upgini-1.1.282a3418.post1 → upgini-1.1.282a3418.post2}/src/upgini/autofe/feature.py

@@ -1,6 +1,6 @@
 import hashlib
 import itertools
-from typing import Dict, List, Optional, Tuple, Union
+from typing import Dict, List, Optional, Set, Tuple, Union
 
 import numpy as np
 import pandas as pd
@@ -94,6 +94,11 @@ class Feature:
         self.alias = alias
         return self
 
+    def get_all_operand_names(self) -> Set[str]:
+        return {self.op.name}.union(
+            {n for f in self.children if isinstance(f, Feature) for n in f.get_all_operand_names()}
+        )
+
     def rename_columns(self, mapping: Dict[str, str]) -> "Feature":
         for child in self.children:
             child.rename_columns(mapping)

{upgini-1.1.282a3418.post1 → upgini-1.1.282a3418.post2}/src/upgini/features_enricher.py

@@ -1442,15 +1442,12 @@ class FeaturesEnricher(TransformerMixin):
             if len(decimal_columns_to_fix) > 0:
                 for col in decimal_columns_to_fix:
                     fitting_eval_X[col] = (
-                        fitting_eval_X[col]
-                        .astype("string").str
-                        .replace(",", ".", regex=False)
-                        .astype(np.float64)
+                        fitting_eval_X[col].astype("string").str.replace(",", ".", regex=False).astype(np.float64)
                     )
                     fitting_enriched_eval_X[col] = (
                         fitting_enriched_eval_X[col]
-                        .astype("string").str
-                        .replace(",", ".", regex=False)
+                        .astype("string")
+                        .str.replace(",", ".", regex=False)
                         .astype(np.float64)
                     )
 
@@ -3293,7 +3290,7 @@ class FeaturesEnricher(TransformerMixin):
                     description[f"Feature {feature_idx}"] = bc.hashed_name
                     feature_idx += 1
 
-                description["Function"] = autofe_feature.op.name
+                description["Function"] = ",".join(sorted(autofe_feature.get_all_operand_names()))
 
                 descriptions.append(description)
 

upgini-1.1.282a3418.post1/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.282a3418-1"