upgini 1.2.103a2__py3-none-any.whl → 1.2.103a4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. upgini/__about__.py +1 -1
  2. upgini/autofe/vector.py +5 -5
  3. {upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/METADATA +1 -1
  4. {upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/RECORD +6 -6
  5. {upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/WHEEL +0 -0
  6. {upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/licenses/LICENSE +0 -0
upgini/__about__.py CHANGED
@@ -1 +1 @@
1
- __version__ = "1.2.103a2"
1
+ __version__ = "1.2.103a4"
upgini/autofe/vector.py CHANGED
@@ -2,7 +2,7 @@ from typing import List, Optional
2
2
 
3
3
  import pandas as pd
4
4
 
5
- from upgini.autofe.operator import OperatorRegistry, PandasOperator, VectorizableMixin
5
+ from upgini.autofe.operator import OperatorRegistry, PandasOperator, ParametrizedOperator, VectorizableMixin
6
6
 
7
7
 
8
8
  class Mean(PandasOperator, VectorizableMixin):
@@ -33,7 +33,7 @@ class Vectorize(PandasOperator, VectorizableMixin):
33
33
  return pd.DataFrame(data).T.apply(lambda x: x.to_list(), axis=1)
34
34
 
35
35
 
36
- class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
36
+ class OnnxModel(PandasOperator, ParametrizedOperator, metaclass=OperatorRegistry):
37
37
  name: str = "onnx"
38
38
  score_name: str = "score"
39
39
  is_vector: bool = True
@@ -43,14 +43,14 @@ class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
43
43
  return f"onnx_{self.score_name}"
44
44
 
45
45
  @classmethod
46
- def from_formula(cls, formula: str) -> "OnnxModel" | None:
46
+ def from_formula(cls, formula: str) -> Optional["OnnxModel"]:
47
47
  if formula.startswith("onnx_"):
48
48
  score_name = formula[len("onnx_"):]
49
49
  return cls(score_name=score_name)
50
50
  return None
51
51
 
52
52
 
53
- class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
53
+ class CatboostModel(PandasOperator, ParametrizedOperator, metaclass=OperatorRegistry):
54
54
  name: str = "catboost"
55
55
  score_name: str = "score"
56
56
  is_vector: bool = True
@@ -60,7 +60,7 @@ class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
60
60
  return f"catboost_{self.score_name}"
61
61
 
62
62
  @classmethod
63
- def from_formula(cls, formula: str) -> "CatboostModel" | None:
63
+ def from_formula(cls, formula: str) -> Optional["CatboostModel"]:
64
64
  if formula.startswith("catboost_"):
65
65
  score_name = formula[len("catboost_"):]
66
66
  return cls(score_name=score_name)
{upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.103a2
3
+ Version: 1.2.103a4
4
4
  Summary: Intelligent data search & enrichment for Machine Learning
5
5
  Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
6
6
  Project-URL: Homepage, https://upgini.com/
{upgini-1.2.103a2.dist-info → upgini-1.2.103a4.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- upgini/__about__.py,sha256=J3yGmfp5EoLlXAMxAKhlLpt4AzwNCq6brJ7PnQanBLM,26
1
+ upgini/__about__.py,sha256=yHthWgrgGwHgGcXGLojFBUHkioApm4qVGSEMkGuW0Pg,26
2
2
  upgini/__init__.py,sha256=LXSfTNU0HnlOkE69VCxkgIKDhWP-JFo_eBQ71OxTr5Y,261
3
3
  upgini/ads.py,sha256=nvuRxRx5MHDMgPr9SiU-fsqRdFaBv8p4_v1oqiysKpc,2714
4
4
  upgini/dataset.py,sha256=e6JDYTZ2AwC5aF-dqclKZKkiKrHo2f6cFmMQO2ZZmjM,32724
@@ -21,7 +21,7 @@ upgini/autofe/groupby.py,sha256=IYmQV9uoCdRcpkeWZj_kI3ObzoNCNx3ff3h8sTL01tk,3603
21
21
  upgini/autofe/operator.py,sha256=RB3rKMjFi5Cx81RiYXN3OTCuXjmvzmFKQrxn4h0Oclo,5219
22
22
  upgini/autofe/unary.py,sha256=FFtvkQaT0cu_zPZ1jCLcsjik-UUh12qQFF3tUW8NqsE,6675
23
23
  upgini/autofe/utils.py,sha256=dYrtyAM8Vcc_R8u4dNo54IsGrHKagTHDJTKhGho0bRg,2967
24
- upgini/autofe/vector.py,sha256=0fvuKj2YSClgDPhKFUoy0mXYoAzCaJiLbv768PIvb74,1973
24
+ upgini/autofe/vector.py,sha256=7GCbXsZv3SYT2pH2DhyQwBoQUadOgaGHUlDkIpgErmo,2045
25
25
  upgini/autofe/timeseries/__init__.py,sha256=PGwwDAMwvkXl3el12tXVEmZUgDUvlmIPlXtROm6bD18,738
26
26
  upgini/autofe/timeseries/base.py,sha256=rWJqRuFAzTZEsUdWG5s1Vhif9zzRRmalASXvarufRxI,3610
27
27
  upgini/autofe/timeseries/cross.py,sha256=BTINVwuZSbm_4NKkVm0FGM68SrvZLENZKXN7-UyvhYI,5319
@@ -71,7 +71,7 @@ upgini/utils/target_utils.py,sha256=i3Xt5l9ybB2_nF_ma5cfPuL3OeFTs2dY2xDI0p4Azpg,
71
71
  upgini/utils/track_info.py,sha256=G5Lu1xxakg2_TQjKZk4b5SvrHsATTXNVV3NbvWtT8k8,5663
72
72
  upgini/utils/ts_utils.py,sha256=26vhC0pN7vLXK6R09EEkMK3Lwb9IVPH7LRdqFIQ3kPs,1383
73
73
  upgini/utils/warning_counter.py,sha256=-GRY8EUggEBKODPSuXAkHn9KnEQwAORC0mmz_tim-PM,254
74
- upgini-1.2.103a2.dist-info/METADATA,sha256=KW8g7N8FTJdaMEkr5I5jTpNqoaNDuts6I4P7x7ySvHc,49531
75
- upgini-1.2.103a2.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
76
- upgini-1.2.103a2.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
77
- upgini-1.2.103a2.dist-info/RECORD,,
74
+ upgini-1.2.103a4.dist-info/METADATA,sha256=YWbq7ZcE4a6fRS4eewE1Pmuzco9LwKUbAF3X3rBMisk,49531
75
+ upgini-1.2.103a4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
76
+ upgini-1.2.103a4.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
77
+ upgini-1.2.103a4.dist-info/RECORD,,