upgini 1.2.103a2__tar.gz → 1.2.103a4__tar.gz

{upgini-1.2.103a2 → upgini-1.2.103a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.103a2
+Version: 1.2.103a4
 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.103a4/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.2.103a4"

{upgini-1.2.103a2 → upgini-1.2.103a4}/src/upgini/autofe/vector.py

@@ -2,7 +2,7 @@ from typing import List, Optional
 
 import pandas as pd
 
-from upgini.autofe.operator import OperatorRegistry, PandasOperator, VectorizableMixin
+from upgini.autofe.operator import OperatorRegistry, PandasOperator, ParametrizedOperator, VectorizableMixin
 
 
 class Mean(PandasOperator, VectorizableMixin):
@@ -33,7 +33,7 @@ class Vectorize(PandasOperator, VectorizableMixin):
         return pd.DataFrame(data).T.apply(lambda x: x.to_list(), axis=1)
 
 
-class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
+class OnnxModel(PandasOperator, ParametrizedOperator, metaclass=OperatorRegistry):
     name: str = "onnx"
     score_name: str = "score"
     is_vector: bool = True
@@ -43,14 +43,14 @@ class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
         return f"onnx_{self.score_name}"
 
     @classmethod
-    def from_formula(cls, formula: str) -> "OnnxModel" | None:
+    def from_formula(cls, formula: str) -> Optional["OnnxModel"]:
         if formula.startswith("onnx_"):
             score_name = formula[len("onnx_"):]
             return cls(score_name=score_name)
         return None
 
 
-class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
+class CatboostModel(PandasOperator, ParametrizedOperator, metaclass=OperatorRegistry):
     name: str = "catboost"
     score_name: str = "score"
     is_vector: bool = True
@@ -60,7 +60,7 @@ class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
         return f"catboost_{self.score_name}"
 
     @classmethod
-    def from_formula(cls, formula: str) -> "CatboostModel" | None:
+    def from_formula(cls, formula: str) -> Optional["CatboostModel"]:
         if formula.startswith("catboost_"):
             score_name = formula[len("catboost_"):]
             return cls(score_name=score_name)

upgini-1.2.103a2/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.2.103a2"