upgini 1.1.280a3418.post9__tar.gz → 1.1.280a3418.post10__tar.gz

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

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

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

{upgini-1.1.280a3418.post9 → upgini-1.1.280a3418.post10}/src/upgini/autofe/date.py

@@ -28,7 +28,7 @@ class DateDiff(PandasOperand, DateDiffMixin):
     is_binary = True
     has_symmetry_importance = True
 
-    def get_params(self) -> Dict[str, str]:
+    def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()
         res.update(
             {
@@ -54,7 +54,7 @@ class DateDiffType2(PandasOperand, DateDiffMixin):
     is_binary = True
     has_symmetry_importance = True
 
-    def get_params(self) -> Dict[str, str]:
+    def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()
         res.update(
             {
@@ -87,7 +87,7 @@ class DateListDiff(PandasOperand, DateDiffMixin):
     has_symmetry_importance = True
     aggregation: str
 
-    def get_params(self) -> Dict[str, str]:
+    def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()
         res.update(
             {
@@ -161,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-1.1.280a3418.post9 → upgini-1.1.280a3418.post10}/src/upgini/autofe/feature.py

@@ -75,8 +75,12 @@ 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

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

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

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

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