upgini 1.1.280a3418.post8__tar.gz → 1.1.280a3418.post9__tar.gz

{upgini-1.1.280a3418.post8 → upgini-1.1.280a3418.post9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.280a3418.post8
+Version: 1.1.280a3418.post9
 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.post9/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.280a3418-9"

{upgini-1.1.280a3418.post8 → upgini-1.1.280a3418.post9}/src/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, 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, 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, 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')}"

{upgini-1.1.280a3418.post8 → upgini-1.1.280a3418.post9}/src/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
@@ -74,6 +77,8 @@ class Feature:
 
     def set_op_params(self, params: Dict[str, str]) -> "Feature":
         self.op.set_params(params)
+        for child in self.children:
+            child.set_op_params(params)
         return self
 
     def get_hash(self) -> str:

{upgini-1.1.280a3418.post8 → upgini-1.1.280a3418.post9}/src/upgini/autofe/operand.py

@@ -26,7 +26,7 @@ class Operand(BaseModel):
         return self
 
     def get_params(self) -> Dict[str, str]:
-        res = {"alias": self.alias}
+        res = {"alias": self.alias} if self.alias is not None else {}
         res.update(self.params or {})
         return res
 

upgini-1.1.280a3418.post8/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.280a3418-8"