upgini 1.1.296a3521.dev5__tar.gz → 1.1.296a3521.dev7__tar.gz

{upgini-1.1.296a3521.dev5 → upgini-1.1.296a3521.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.296a3521.dev5
+Version: 1.1.296a3521.dev7
 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.296a3521.dev7/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.296a3521.dev7"

{upgini-1.1.296a3521.dev5 → upgini-1.1.296a3521.dev7}/src/upgini/autofe/date.py

@@ -185,6 +185,15 @@ class DatePercentileBase(PandasOperand, abc.ABC):
         else:
             return np.nan
 
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res.update(
+            {
+                "date_unit": self.date_unit,
+            }
+        )
+        return res
+
 
 class DatePercentile(DatePercentileBase):
     name = "date_per"
@@ -198,7 +207,6 @@ class DatePercentile(DatePercentileBase):
         res = super().get_params()
         res.update(
             {
-                "date_unit": self.date_unit,
                 "zero_month": self.zero_month,
                 "zero_year": self.zero_year,
                 "zero_bounds": self.zero_bounds,

{upgini-1.1.296a3521.dev5 → upgini-1.1.296a3521.dev7}/src/upgini/autofe/feature.py

@@ -325,17 +325,18 @@ class FeatureGroup:
             lower_order_children = [
                 ch for f in self.children for ch in f.children if ch.get_display_name() != main_column
             ]
+            lower_order_names = [ch.get_display_name() for ch in lower_order_children]
             if any(isinstance(f, Feature) for f in lower_order_children):
                 child_data = pd.concat(
-                    [data[main_column]] + [ch.calculate(data) for ch in lower_order_children], axis=1
+                    [data[main_column]] + [ch.calculate(data) for ch in lower_order_children],
+                    axis=1,
                 )
+                child_data.columns = [main_column] + lower_order_names
             else:
                 child_data = data[columns]
 
             new_data = self.op.calculate_group(child_data, main_column=main_column)
-            new_data.rename(
-                columns=dict(zip((c for c in columns if c != main_column), self.get_display_names())), inplace=True
-            )
+            new_data.rename(columns=dict(zip(lower_order_names, self.get_display_names())), inplace=True)
         else:
             raise NotImplementedError(f"Unrecognized operator {self.op.name}.")
 

upgini-1.1.296a3521.dev5/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.296a3521.dev5"