upgini 1.2.103a3__py3-none-any.whl → 1.2.103a5__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 +7 -3
  3. {upgini-1.2.103a3.dist-info → upgini-1.2.103a5.dist-info}/METADATA +1 -1
  4. {upgini-1.2.103a3.dist-info → upgini-1.2.103a5.dist-info}/RECORD +6 -6
  5. {upgini-1.2.103a3.dist-info → upgini-1.2.103a5.dist-info}/WHEEL +0 -0
  6. {upgini-1.2.103a3.dist-info → upgini-1.2.103a5.dist-info}/licenses/LICENSE +0 -0
upgini/__about__.py CHANGED
@@ -1 +1 @@
1
- __version__ = "1.2.103a3"
1
+ __version__ = "1.2.103a5"
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
@@ -47,10 +47,12 @@ class OnnxModel(PandasOperator, metaclass=OperatorRegistry):
47
47
  if formula.startswith("onnx_"):
48
48
  score_name = formula[len("onnx_"):]
49
49
  return cls(score_name=score_name)
50
+ elif formula == "onnx": # for OperatorRegistry
51
+ return cls()
50
52
  return None
51
53
 
52
54
 
53
- class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
55
+ class CatboostModel(PandasOperator, ParametrizedOperator, metaclass=OperatorRegistry):
54
56
  name: str = "catboost"
55
57
  score_name: str = "score"
56
58
  is_vector: bool = True
@@ -64,4 +66,6 @@ class CatboostModel(PandasOperator, metaclass=OperatorRegistry):
64
66
  if formula.startswith("catboost_"):
65
67
  score_name = formula[len("catboost_"):]
66
68
  return cls(score_name=score_name)
69
+ elif formula == "catboost": # for OperatorRegistry
70
+ return cls()
67
71
  return None
{upgini-1.2.103a3.dist-info → upgini-1.2.103a5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: upgini
3
- Version: 1.2.103a3
3
+ Version: 1.2.103a5
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.103a3.dist-info → upgini-1.2.103a5.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- upgini/__about__.py,sha256=jDLHCCktfLiSVYr9xsHCkBKrGQk6hR0nkymrhbE64ek,26
1
+ upgini/__about__.py,sha256=0-FPlqudOzhknjOVHM69SGhLLu6ORJxXIWxqnNFRH1I,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=I0T0WUsuOdabWdaFfGgohxO65hhlXe7zqVXBSFrkUv4,1979
24
+ upgini/autofe/vector.py,sha256=tsKDTRP-tbBl5_BNHr3zsciX6Kt7ymOUJfzUpRPVOMQ,2211
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.103a3.dist-info/METADATA,sha256=C4-tS0gzS2ON0MUqVWbBkJW5PKygS3KbfiE8IinUqII,49531
75
- upgini-1.2.103a3.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
76
- upgini-1.2.103a3.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
77
- upgini-1.2.103a3.dist-info/RECORD,,
74
+ upgini-1.2.103a5.dist-info/METADATA,sha256=coktUdKKQ7jLimcxPEPbMlp27cNuOyoHmSp-VHvOsEU,49531
75
+ upgini-1.2.103a5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
76
+ upgini-1.2.103a5.dist-info/licenses/LICENSE,sha256=5RRzgvdJUu3BUDfv4bzVU6FqKgwHlIay63pPCSmSgzw,1514
77
+ upgini-1.2.103a5.dist-info/RECORD,,