upgini 1.2.102a1__py3-none-any.whl → 1.2.103a2__py3-none-any.whl

upgini/__about__.py

@@ -1 +1 @@
-__version__ = "1.2.102a1"
+__version__ = "1.2.103a2"

upgini/autofe/vector.py

@@ -35,11 +35,33 @@ class Vectorize(PandasOperator, VectorizableMixin):
 
 class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
     name: str = "onnx"
+    score_name: str = "score"
     is_vector: bool = True
     output_type: Optional[str] = "float"
 
+    def to_formula(self) -> str:
+        return f"onnx_{self.score_name}"
+
+    @classmethod
+    def from_formula(cls, formula: str) -> "OnnxModel" | None:
+        if formula.startswith("onnx_"):
+            score_name = formula[len("onnx_"):]
+            return cls(score_name=score_name)
+        return None
+
 
 class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
     name: str = "catboost"
+    score_name: str = "score"
     is_vector: bool = True
     output_type: Optional[str] = "float"
+
+    def to_formula(self) -> str:
+        return f"catboost_{self.score_name}"
+
+    @classmethod
+    def from_formula(cls, formula: str) -> "CatboostModel" | None:
+        if formula.startswith("catboost_"):
+            score_name = formula[len("catboost_"):]
+            return cls(score_name=score_name)
+        return None

{upgini-1.2.102a1.dist-info → upgini-1.2.103a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.2.102a1
+Version: 1.2.103a2
 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.102a1.dist-info → upgini-1.2.103a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-upgini/__about__.py,sha256=I6O0u_ro3Zy1sT3c17lfXXjinJn-P0VMoCHC5aC0ft4,26
+upgini/__about__.py,sha256=J3yGmfp5EoLlXAMxAKhlLpt4AzwNCq6brJ7PnQanBLM,26
 upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
 upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
 upgini/dataset.py,sha256=e6JDYTZ2AwC5aF-dqclKZKkiKrHo2f6cFmMQO2ZZmjM,32724
@@ -21,7 +21,7 @@ upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
 upgini/autofe/operator.py,sha256=RB3rKMjFi5Cx81RiYXN3OTCuXjmvzmFKQrxn4h0Oclo,5219
 upgini/autofe/unary.py,sha256=FFtvkQaT0cu_zPZ1jCLcsjik-UUh12qQFF3tUW8NqsE,6675
 upgini/autofe/utils.py,sha256=dYrtyAM8Vcc_R8u4dNo54IsGrHKagTHDJTKhGho0bRg,2967
-upgini/autofe/vector.py,sha256=9T7MEUK0SavXIJy0c9Kvu5qTcMtt3fzvdRDBDxcI0JA,1277
+upgini/autofe/vector.py,sha256=0fvuKj2YSClgDPhKFUoy0mXYoAzCaJiLbv768PIvb74,1973
 upgini/autofe/timeseries/__init__.py,sha256=PGwwDAMwvkXl3el12tXVEmZUgDUvlmIPlXtROm6bD18,738
 upgini/autofe/timeseries/base.py,sha256=rWJqRuFAzTZEsUdWG5s1Vhif9zzRRmalASXvarufRxI,3610
 upgini/autofe/timeseries/cross.py,sha256=BTINVwuZSbm_4NKkVm0FGM68SrvZLENZKXN7-UyvhYI,5319
@@ -71,7 +71,7 @@ upgini/utils/target_utils.py,sha256=i3Xt5l9ybB2_nF_ma5cfPuL3OeFTs2dY2xDI0p4Azpg,
 upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
 upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
 upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
-upgini-1.2.102a1.dist-info/METADATA,sha256=WppwVWMFj8biE0bf1NZwLLUXvfmZkiKzGke-ScyWzFE,49531
-upgini-1.2.102a1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-upgini-1.2.102a1.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
-upgini-1.2.102a1.dist-info/RECORD,,
+upgini-1.2.103a2.dist-info/METADATA,sha256=KW8g7N8FTJdaMEkr5I5jTpNqoaNDuts6I4P7x7ySvHc,49531
+upgini-1.2.103a2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+upgini-1.2.103a2.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
+upgini-1.2.103a2.dist-info/RECORD,,