autogluon.core 1.2.1b20250305__py3-none-any.whl → 1.2.1b20250306__py3-none-any.whl

autogluon/core/models/abstract/abstract_model.py

@@ -128,7 +128,7 @@ class Tunable(ABC):
         pass
 
 
-class ModelBase(Taggable, Tunable, ABC):
+class ModelBase(Taggable, ABC):
     @abstractmethod
     def __init__(
         self,
@@ -166,7 +166,7 @@ class ModelBase(Taggable, Tunable, ABC):
 
 
 # TODO: move to tabular, rename AbstractTabularModel
-class AbstractModel(ModelBase):
+class AbstractModel(ModelBase, Tunable):
     """
     Abstract model implementation from which all AutoGluon models inherit.
 
@@ -1676,11 +1676,13 @@ class AbstractModel(ModelBase):
         """
         return [((batch_size, len(self._features)), np.float32)]
 
-    def _default_compiler(self):
+    @classmethod
+    def _default_compiler(cls):
         """The default compiler for the underlining model."""
         return None
 
-    def _valid_compilers(self) -> list:
+    @classmethod
+    def _valid_compilers(cls) -> list:
         """A list of supported compilers for the underlining model."""
         return []
 
@@ -2649,3 +2651,7 @@ class AbstractModel(ModelBase):
             return cls.ag_priority
         else:
             return cls.ag_priority_by_problem_type.get(problem_type, cls.ag_priority)
+
+    @classmethod
+    def _class_tags(cls) -> dict:
+        return {"supports_learning_curves": False}

autogluon/core/models/dummy/dummy_model.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas as pd
 
 from ...constants import BINARY, MULTICLASS, QUANTILE, REGRESSION

autogluon/core/models/ensemble/weighted_ensemble_model.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from collections import defaultdict
 

autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 
 from autogluon.common.features.types import S_STACK
@@ -126,6 +128,10 @@ class GreedyWeightedEnsembleModel(AbstractModel):
         default_ag_args.update(extra_ag_args)
         return default_ag_args
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile", "softclass"]
+
     def _get_default_stopping_metric(self):
         return self.eval_metric
 

autogluon/core/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250305"
+__version__ = "1.2.1b20250306"
 __lite__ = False

{autogluon.core-1.2.1b20250305.dist-info → autogluon.core-1.2.1b20250306.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250305
+Version: 1.2.1b20250306
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -45,11 +45,11 @@ Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: requests
 Requires-Dist: matplotlib<3.11,>=3.7.0
 Requires-Dist: boto3<2,>=1.10
-Requires-Dist: autogluon.common==1.2.1b20250305
+Requires-Dist: autogluon.common==1.2.1b20250306
 Provides-Extra: all
-Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "all"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
 Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "all"
+Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "all"
 Requires-Dist: pyarrow>=15.0.0; extra == "all"
 Provides-Extra: ray
 Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "ray"
@@ -59,10 +59,10 @@ Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "raytune"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
-Requires-Dist: pytest-mypy; extra == "tests"
-Requires-Dist: flake8; extra == "tests"
 Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
+Requires-Dist: flake8; extra == "tests"
+Requires-Dist: pytest-mypy; extra == "tests"
 
 
 

{autogluon.core-1.2.1b20250305.dist-info → autogluon.core-1.2.1b20250306.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250305-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250306-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
 autogluon/core/_setup_utils.py,sha256=ikxn4zc3PNyjJJT5SsgL0dvP-6Rbq6_dItGMiZNINv4,6958
 autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
 autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
-autogluon/core/version.py,sha256=hjp__CpU1tLt2wGIEfd4tVSKrES_x0vHTkZdLuoYGnE,91
+autogluon/core/version.py,sha256=vpLPQw8Tz1FiEFDl7STt_ZrtxXTVYzhhywUGcg_8Fec,91
 autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
 autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
@@ -40,21 +40,21 @@ autogluon/core/models/__init__.py,sha256=raRAG4wml-xvCKp3zxthda7of9NEWcpafgpvZWJ
 autogluon/core/models/_utils.py,sha256=qswE9n1ge1AJSExgstEbrZiMFmMRa4Mf5Sz8D9-XU6c,2091
 autogluon/core/models/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/abstract/_tags.py,sha256=Qr_3an0ZMig24S3OwISa-nTFfWHQe3pwPTiXq4zlEec,3409
-autogluon/core/models/abstract/abstract_model.py,sha256=MaCggM7TwX5y9TyOQNV0N4cfnKmcY6kWyf2fUmDthkA,124174
+autogluon/core/models/abstract/abstract_model.py,sha256=MgR0uB3hFQFM-NaHRBhg9e1gWNlu-zHEm2_wURWtHLs,124309
 autogluon/core/models/abstract/abstract_nn_model.py,sha256=IId0ivO8uVvmpnK9OiM2CtPVrP1ewOaQQKtQUDtK7_k,4818
 autogluon/core/models/abstract/model_trial.py,sha256=PKEo1jfLSBCOLM42QE5VBD1u41MaVMRk31zhNhLiqTw,5035
 autogluon/core/models/dummy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/dummy/_dummy_quantile_regressor.py,sha256=i-ZW2flJ60jsMfMK24IP39Xwc55-UlBDvHmqanIf29Q,664
-autogluon/core/models/dummy/dummy_model.py,sha256=VfIIvEQrX1n8HktUGiMIG_GeNzdDgiwysJ3M-236JxU,1599
+autogluon/core/models/dummy/dummy_model.py,sha256=oCAL0HnPJEREPtgnS_N8YtCdn99yXW0zD8EBEJJToNA,1635
 autogluon/core/models/ensemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/ensemble/bagged_ensemble_model.py,sha256=kQbJhq_UzUoOHsTX00WTEXu1Ogdme4bdLiiWhIMNUfU,77257
 autogluon/core/models/ensemble/fold_fitting_strategy.py,sha256=01vzNVvE4FIFgD6YqbhK63XoUlSztnVFsrDdsoqm75U,47021
 autogluon/core/models/ensemble/ray_parallel_fold_fitting_strategy.py,sha256=8RASa-eV6n9kUgbqQHNt7k4IrvuB9NdrunIMLYOLwgA,2068
 autogluon/core/models/ensemble/stacker_ensemble_model.py,sha256=oGgn0oxu1jzygcCHKBvlMZLbKrDAIgcrOMXFbSTNTVM,18059
-autogluon/core/models/ensemble/weighted_ensemble_model.py,sha256=hELT3L7tDRYuN7Ghm_9zNH4zzG_g8vk0hINofu_ZqSE,4318
+autogluon/core/models/ensemble/weighted_ensemble_model.py,sha256=6FOLKJbk3FcFapsDlw7XY5KHaG8jUwjl0V_Uj4wQsH4,4354
 autogluon/core/models/greedy_ensemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/greedy_ensemble/ensemble_selection.py,sha256=Rt5jkqIORaCFLiQlh-LVWKP5-eB6gxEnVNmKTB3RP58,10548
-autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py,sha256=XrNiIRbU-XXICSF_m8G60TRFLauwXu9MQgvr2JbOD3E,7172
+autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py,sha256=pHpVhBCqe9_jNFq812HGiYT2pHv_AzMPMfQFtXvm7HU,7363
 autogluon/core/pseudolabeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/pseudolabeling/pseudolabeling.py,sha256=vvOrY02Wvba3UGujYH3xqgI6yf2jRegzLiNfKrxSais,12138
 autogluon/core/ray/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
 autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
 autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
 autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
-autogluon.core-1.2.1b20250305.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250305.dist-info/METADATA,sha256=J0uWw9lsHc1-NWm4mzvexNOz23Dkc8gkupllboscgvk,12399
-autogluon.core-1.2.1b20250305.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250305.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250305.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250305.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250305.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250305.dist-info/RECORD,,
+autogluon.core-1.2.1b20250306.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250306.dist-info/METADATA,sha256=zu01FR8M6niA3jS9iV6e40t2WAi0VJPAxm8JY53nsBU,12399
+autogluon.core-1.2.1b20250306.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250306.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250306.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250306.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250306.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250306.dist-info/RECORD,,