upgini 1.1.296a3521.dev4__py3-none-any.whl → 1.1.296a3521.dev5__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.296a3521.dev4"
+__version__ = "1.1.296a3521.dev5"

upgini/autofe/feature.py

@@ -16,6 +16,9 @@ class Column:
         self.data = data
         self.calculate_all = calculate_all
 
+    def get_name_component(self, **kwargs) -> str:
+        return self.name
+
     def get_display_name(self, cache: bool = True, shorten: bool = False, **kwargs) -> str:
         return self.name
 
@@ -41,7 +44,7 @@ class Column:
     def get_column_nodes(self) -> List["Column"]:
         return [self]
 
-    def get_columns(self) -> List[str]:
+    def get_columns(self, **kwargs) -> List[str]:
         return [self.name]
 
     def infer_type(self, data: pd.DataFrame) -> DtypeObj:
@@ -57,6 +60,12 @@ class Column:
     def to_pretty_formula(self) -> str:
         return self.to_formula()
 
+    def __eq__(self, value: object) -> bool:
+        if not isinstance(value, Column):
+            return False
+        else:
+            return self.name == value.name and self.calculate_all == value.calculate_all
+
 
 class Feature:
     def __init__(
@@ -125,6 +134,12 @@ class Feature:
         for child in self.children:
             child.delete_data()
 
+    def get_name_component(self, **kwargs) -> str:
+        return "_".join(ch.get_name_component(**kwargs) for ch in self.children) + "_" + self.get_op_display_name()
+
+    def get_op_display_name(self) -> str:
+        return self.op.alias or self.op.name.lower()
+
     def get_display_name(self, cache: bool = True, shorten: bool = False, **kwargs) -> str:
         if self.cached_display_name is not None and cache:
             return self.cached_display_name
@@ -132,11 +147,12 @@ class Feature:
         if self.alias:
             components = ["f_autofe", self.alias]
         elif shorten and not self.op.is_unary:
-            components = ["f_autofe", self.op.alias or self.op.name.lower()]
+            components = ["f_autofe", self.get_op_display_name()]
         else:
-            components = ["f_" + "_f_".join(self.get_columns(**kwargs))] + [
+            child_components = [ch.get_name_component(**kwargs) for ch in self.children]
+            components = ["f_" + "_f_".join(child_components)] + [
                 "autofe",
-                self.op.alias or self.op.name.lower(),
+                self.get_op_display_name(),
             ]
         components.extend([str(self.display_index)] if self.display_index is not None else [])
         display_name = "_".join(components)
@@ -306,8 +322,20 @@ class FeatureGroup:
         main_column = None if self.main_column_node is None else self.main_column_node.get_columns()[0]
         if isinstance(self.op, PandasOperand):
             columns = self.get_columns()
-            new_data = self.op.calculate_group(data[columns], main_column=main_column)
-            new_data.rename(columns=dict(zip(columns, self.get_display_names())), inplace=True)
+            lower_order_children = [
+                ch for f in self.children for ch in f.children if ch.get_display_name() != main_column
+            ]
+            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
+                )
+            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
+            )
         else:
             raise NotImplementedError(f"Unrecognized operator {self.op.name}.")
 

{upgini-1.1.296a3521.dev4.dist-info → upgini-1.1.296a3521.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.296a3521.dev4
+Version: 1.1.296a3521.dev5
 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.dev4.dist-info → upgini-1.1.296a3521.dev5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=VGHsR03QeKC0Gtu5pTlpKTuNsj9FK3EAS6Xjs7vhoYs,34
+upgini/__about__.py,sha256=hw_DSyyM8S283cgz_zkPjXawAFKkZq2W9BY3bjwwNhE,34
 upgini/__init__.py,sha256=ObEtjFkIssl83qeKNMLpIQygfwK8TzztwiI43YTsAP0,353
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=7TLVVhGtjgx_9yaiaIUK3kZSe_R9wg5dY0d4F5qCGM4,45636
@@ -17,7 +17,7 @@ upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operands.py,sha256=WJOiNVNVyrWc3vYGMvJtrpGRrLEQ237YMTsWnkUbLNw,2502
 upgini/autofe/binary.py,sha256=8FXPJxN7fnC5wphO0Dp1tQCa0lFMSDGQGvBMkSIVAcE,4155
 upgini/autofe/date.py,sha256=va28wBqELETjn1CqZ7bHvz5CcJg8sijFyo9Ti7P-HLw,7711
-upgini/autofe/feature.py,sha256=_V9B74B3ue7eAYXSOt9JKhVC9klkAKks22MwnBRye_w,12487
+upgini/autofe/feature.py,sha256=TbPW81696BMbxwxyk14Jv_98kKqDbSKbVK62pjkfI1A,13621
 upgini/autofe/groupby.py,sha256=4WjDzQxqpZxB79Ih4ihMMI5GDxaFqiH6ZelfV82ClT4,3091
 upgini/autofe/operand.py,sha256=MKEsl3zxpWzRDpTkE0sNJxTu62U20sWOvEKhPjUWS6s,2915
 upgini/autofe/unary.py,sha256=ZWjLd-CUkNt_PpM8YuWLLipW1v_RdBlsl4JxXIVo9aM,3652
@@ -57,7 +57,7 @@ upgini/utils/sklearn_ext.py,sha256=13jQS_k7v0aUtudXV6nGUEWjttPQzAW9AFYL5wgEz9k,4
 upgini/utils/target_utils.py,sha256=Y96_PJ5cC-WsEbeqg20v9uqywDQobLoTb-xoP7S3o4E,7807
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/warning_counter.py,sha256=dIWBB4dI5XRRJZudvIlqlIYKEiwLLPcXarsZuYRt338,227
-upgini-1.1.296a3521.dev4.dist-info/METADATA,sha256=wb-EbeLcSYxsUnpMDxht1FCJJBeE87CTuwo3RoF-02A,48127
-upgini-1.1.296a3521.dev4.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.1.296a3521.dev4.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.296a3521.dev4.dist-info/RECORD,,
+upgini-1.1.296a3521.dev5.dist-info/METADATA,sha256=mqBWJm9KU8FLVsvBVixKGPCT7DTBLnxuF4qqD6t8kDo,48127
+upgini-1.1.296a3521.dev5.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+upgini-1.1.296a3521.dev5.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.296a3521.dev5.dist-info/RECORD,,