upgini 1.1.280a3418.post8__py3-none-any.whl → 1.1.280a3418.post10__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.1.280a3418-8"
+__version__ = "1.1.280a3418-10"

upgini/autofe/date.py

@@ -28,6 +28,17 @@ class DateDiff(PandasOperand, DateDiffMixin):
     is_binary = True
     has_symmetry_importance = True
 
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res.update(
+            {
+                "diff_unit": self.diff_unit,
+                "left_unit": self.left_unit,
+                "right_unit": self.right_unit,
+            }
+        )
+        return res
+
     def calculate_binary(self, left: pd.Series, right: pd.Series) -> pd.Series:
         left = self._convert_to_date(left, self.left_unit)
         right = self._convert_to_date(right, self.right_unit)
@@ -43,6 +54,17 @@ class DateDiffType2(PandasOperand, DateDiffMixin):
     is_binary = True
     has_symmetry_importance = True
 
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res.update(
+            {
+                "diff_unit": self.diff_unit,
+                "left_unit": self.left_unit,
+                "right_unit": self.right_unit,
+            }
+        )
+        return res
+
     def calculate_binary(self, left: pd.Series, right: pd.Series) -> pd.Series:
         left = self._convert_to_date(left, self.left_unit)
         right = self._convert_to_date(right, self.right_unit)
@@ -65,6 +87,15 @@ class DateListDiff(PandasOperand, DateDiffMixin):
     has_symmetry_importance = True
     aggregation: str
 
+    def get_params(self) -> Dict[str, Optional[str]]:
+        res = super().get_params()
+        res.update(
+            {
+                "aggregation": self.aggregation,
+            }
+        )
+        return res
+
     def __init__(self, **data: Any) -> None:
         if "name" not in data:
             data["name"] = f"date_diff_{data.get('aggregation')}"
@@ -130,7 +161,7 @@ class DatePercentile(PandasOperand):
     zero_bounds: Optional[List[float]]
     step: int = 30
 
-    def get_params(self) -> Dict[str, str]:
+    def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()
         res.update(
             {

upgini/autofe/feature.py

@@ -19,6 +19,9 @@ class Column:
     def get_display_name(self, cache: bool = True, shorten: bool = False, **kwargs) -> str:
         return self.name
 
+    def set_op_params(self, params: Dict[str, str]) -> "Column":
+        return self
+
     def rename_columns(self, mapping: Dict[str, str]) -> "Column":
         self.name = self._unhash(mapping.get(self.name) or self.name)
         return self
@@ -72,8 +75,14 @@ class Feature:
         self.cached_display_name = cached_display_name
         self.alias = alias
 
-    def set_op_params(self, params: Dict[str, str]) -> "Feature":
+    def set_op_params(self, params: Optional[Dict[str, str]]) -> "Feature":
+        obj_dict = self.op.dict().copy()
+        obj_dict.update(params or {})
+        self.op = self.op.__class__.parse_obj(obj_dict)
         self.op.set_params(params)
+
+        for child in self.children:
+            child.set_op_params(params)
         return self
 
     def get_hash(self) -> str:

upgini/autofe/operand.py

@@ -25,7 +25,7 @@ class Operand(BaseModel):
         self.params = params
         return self
 
-    def get_params(self) -> Dict[str, str]:
+    def get_params(self) -> Dict[str, Optional[str]]:
         res = {"alias": self.alias}
         res.update(self.params or {})
         return res

{upgini-1.1.280a3418.post8.dist-info → upgini-1.1.280a3418.post10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.280a3418.post8
+Version: 1.1.280a3418.post10
 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.280a3418.post8.dist-info → upgini-1.1.280a3418.post10.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=N5R3kVOqIy0K6aHAxFEXNCurBV79BBXzp56eDSeAjaY,31
+upgini/__about__.py,sha256=dmfbxwIubpazkuRbZUmqQwuEKtDwBSXJuDdQN0rHZuY,32
 upgini/__init__.py,sha256=asENHgEVHQBIkV-e_0IhE_ZWqkCG6398U3ZLrNzAH6k,407
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=uiFY-P8te7-zigib1hGWRtW5v0X7chxPM0hJFdixAN8,45623
@@ -15,10 +15,10 @@ upgini/ads_management/ads_manager.py,sha256=igVbN2jz80Umb2BUJixmJVj-zx8unoKpecVo
 upgini/autofe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 upgini/autofe/all_operands.py,sha256=7UyvmmqGSqQu4kDgoFwQRKY__b9xKDk3Fpp2-H8A7AA,2399
 upgini/autofe/binary.py,sha256=441BRuqMsxlxuw4c8rMZB6h5EpRdVMk-bVa03U7T5Hg,3973
-upgini/autofe/date.py,sha256=y3l97i_TIBaCJ-MuvQEwk5Ch6ofPsbqPzPSwn1qRS60,5924
-upgini/autofe/feature.py,sha256=8ugRW-Ckc-SYNQawgEluoAzJ_oM5v6y9r5Qx_E-C14o,11977
+upgini/autofe/date.py,sha256=XeTJIkPzauYXgI3n5E8h4353PhZRZ7LOpQGHA4DMPx4,6758
+upgini/autofe/feature.py,sha256=x7sumlIZqSE020XAkoykaesUjmmdYhaa0iFPKxBuDXo,12285
 upgini/autofe/groupby.py,sha256=4WjDzQxqpZxB79Ih4ihMMI5GDxaFqiH6ZelfV82ClT4,3091
-upgini/autofe/operand.py,sha256=RRGEQCFSzeaHqh7KnEWxivbEeCUaKONOYzefO-RpiFY,2872
+upgini/autofe/operand.py,sha256=JjEVT1U3kY9NDjUPMdoki7Oa8hMDG0-_h_NklVjIFyc,2882
 upgini/autofe/unary.py,sha256=v-l3aiE5hj6kurvh6adCQL8W3X9u9a7RVbS_WPR2qlw,3146
 upgini/autofe/vector.py,sha256=dLxfAstJs-gw_OQ1xxoxcM6pVzORlV0HVzdzt7cLXVQ,606
 upgini/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -56,7 +56,7 @@ upgini/utils/sklearn_ext.py,sha256=c23MGSUVfxLnaDWKAxavHgnOtm5dGKkF3YswdWQcFzs,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.280a3418.post8.dist-info/METADATA,sha256=rpY21TutqvORnVgF-piUMYosXzVa9-in-o_EIKybqeU,48129
-upgini-1.1.280a3418.post8.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-upgini-1.1.280a3418.post8.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.1.280a3418.post8.dist-info/RECORD,,
+upgini-1.1.280a3418.post10.dist-info/METADATA,sha256=IYik9d-rA5UCa6NmuR2N8bfOHJq-T-mw2AVzXGfnFyQ,48130
+upgini-1.1.280a3418.post10.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+upgini-1.1.280a3418.post10.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.1.280a3418.post10.dist-info/RECORD,,