upgini 1.2.98a3594.dev1__tar.gz → 1.2.98a3594.dev3__tar.gz

{upgini-1.2.98a3594.dev1 → upgini-1.2.98a3594.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.98a3594.dev1
+Version: 1.2.98a3594.dev3
 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.2.98a3594.dev3/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.98a3594.dev3"

{upgini-1.2.98a3594.dev1 → upgini-1.2.98a3594.dev3}/src/upgini/autofe/unary.py

@@ -3,7 +3,7 @@ from typing import Dict, List, Optional
 import numpy as np
 import pandas as pd
 
-from upgini.autofe.operator import PandasOperator, VectorizableMixin
+from upgini.autofe.operator import PandasOperator, ParametrizedOperator, VectorizableMixin
 from upgini.autofe.utils import pydantic_validator
 
 
@@ -200,12 +200,29 @@ class Cluster(PandasOperator):
     is_categorical: bool = True
 
 
-class OutlierDistance(PandasOperator):
+class OutlierDistance(PandasOperator, ParametrizedOperator):
     name: str = "outlier_dist"
     is_unary: bool = True
     input_type: Optional[str] = "vector"
     output_type: Optional[str] = "float"
     centroid: Optional[List[float]] = None
+    class_value: Optional[str] = None
+
+    def to_formula(self) -> str:
+        return f"outlier_dist_{self.class_value if self.class_value is not None else 'all'}"
+
+    @classmethod
+    def from_formula(cls, formula: str) -> Optional["OutlierDistance"]:
+        if formula.startswith("outlier_dist_"):
+            class_value = formula.split("_")[-1]
+            return cls(class_value=None if class_value == "all" else class_value)
+        return None
+
+    @pydantic_validator("centroid", mode="before")
+    def parse_centroid(cls, value):
+        if isinstance(value, str):
+            return json.loads(value)
+        return value
 
     def get_params(self) -> Dict[str, Optional[str]]:
         res = super().get_params()

upgini-1.2.98a3594.dev1/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.98a3594.dev1"