upgini 1.2.98a3594.dev2__tar.gz → 1.2.98a3594.dev4__tar.gz

{upgini-1.2.98a3594.dev2 → upgini-1.2.98a3594.dev4}/PKG-INFO

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

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

{upgini-1.2.98a3594.dev2 → upgini-1.2.98a3594.dev4}/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,8 +200,22 @@ 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"
+    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
+
+    def get_centroid(self) -> np.ndarray:
+        pass

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

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