autogluon.core 1.2.1b20250228__py3-none-any.whl → 1.2.1b20250301__py3-none-any.whl

autogluon/core/data/label_cleaner.py

@@ -339,3 +339,9 @@ class LabelCleanerDummy(LabelCleaner):
 
     def _inverse_transform(self, y: Union[Series, DataFrame]) -> Union[Series, DataFrame]:
         return y
+
+    def to_transformed_dtype(self, y: Union[Series, np.ndarray, list]) -> Series:
+        return y
+
+    def to_original_dtype(self, y: Union[Series, np.ndarray, list]) -> Series:
+        return y

autogluon/core/models/abstract/abstract_model.py

@@ -2486,6 +2486,9 @@ class AbstractModel(ModelBase):
         """
         Dictionary of customization options related to meta properties of the model such as its name, the order it is trained, and the problem types it is valid for.
         """
+        supported_problem_types = cls.supported_problem_types()
+        if supported_problem_types is not None:
+            return {"problem_types": supported_problem_types}
         return {}
 
     @classmethod
@@ -2496,6 +2499,15 @@ class AbstractModel(ModelBase):
         """
         return {}
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        """
+        Returns the list of supported problem types.
+        If None is returned, then the model has not specified the supported problem types, and it is unknown which problem types are valid.
+            In this case, all problem types are considered supported and the model will never be filtered out based on problem type.
+        """
+        return None
+
     def _get_default_stopping_metric(self) -> Scorer:
         """
         Returns the default stopping metric to use for early stopping.

autogluon/core/models/dummy/dummy_model.py

@@ -40,3 +40,7 @@ class DummyModel(AbstractModel):
         else:
             self.model = model_cls()
         self.model.fit(X=X, y=y)
+
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile"]

autogluon/core/version.py

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

{autogluon.core-1.2.1b20250228.dist-info → autogluon.core-1.2.1b20250301.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250228
+Version: 1.2.1b20250301
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -45,12 +45,12 @@ 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.1b20250228
+Requires-Dist: autogluon.common==1.2.1b20250301
 Provides-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"
+Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "all"
 Provides-Extra: ray
 Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "ray"
 Provides-Extra: raytune
@@ -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: pytest-mypy; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 
 
 

{autogluon.core-1.2.1b20250228.dist-info → autogluon.core-1.2.1b20250301.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250228-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250301-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=4RzEVBLupX6W_qKJDfxEuDLJ2Ppi02arrX2mkTUJanE,91
+autogluon/core/version.py,sha256=6kceKq_ABW7D8n2TxlTDuCVTTOR7a0FEMpqKYUmZerA,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
@@ -17,7 +17,7 @@ autogluon/core/callbacks/_early_stopping_ensemble_callback.py,sha256=lNcIa8GJ88L
 autogluon/core/callbacks/_example_callback.py,sha256=xh4dDe4bECaj24bfLtxZysT7w4SQvcYbONw72EFzCI8,2597
 autogluon/core/data/__init__.py,sha256=BJSosOhkodKk-gC2oqa6Wkkz8ZmiHwy0jZLXBhL3FYE,40
 autogluon/core/data/cleaner.py,sha256=ojwr1BemM8RSmH81uBDhMIJZ2zavXyjGgkA7NKaMbV0,2803
-autogluon/core/data/label_cleaner.py,sha256=f_K7cEHMBdQek8EhkbRD1SV2R8i4K2eGis-VsRIsMbg,14904
+autogluon/core/data/label_cleaner.py,sha256=rgXFICs3oNlkaglnSohwjh-vcSaN2mcJTDAJAnBjSPQ,15101
 autogluon/core/hpo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/hpo/constants.py,sha256=vYeJNySZtldqKX3hctMM-RMT8UuVZneH3tAneiIaIZM,117
 autogluon/core/hpo/exceptions.py,sha256=TvkCQZSd-TMwIqmxkOAONMuiz-HimxwHwjvGynOj5ug,44
@@ -40,12 +40,12 @@ 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=8FhnDpjLN_32O9qqVk_ZR7eu5o1djbl4UYVyCbrX7eo,123554
+autogluon/core/models/abstract/abstract_model.py,sha256=MaCggM7TwX5y9TyOQNV0N4cfnKmcY6kWyf2fUmDthkA,124174
 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=2dDGv8kShxrJPlCcthTj3dCZ6AvGFtB2knoi4x8qhCc,1457
+autogluon/core/models/dummy/dummy_model.py,sha256=VfIIvEQrX1n8HktUGiMIG_GeNzdDgiwysJ3M-236JxU,1599
 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
@@ -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.1b20250228.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250228.dist-info/METADATA,sha256=pnhxaHZeMdljD94ytDy7XvQ-jVH_qSP3VduiQ0YqqTQ,12399
-autogluon.core-1.2.1b20250228.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250228.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250228.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250228.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250228.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250228.dist-info/RECORD,,
+autogluon.core-1.2.1b20250301.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250301.dist-info/METADATA,sha256=cryYvmIvTssgnGw9XsAsnazRaDZEWJORguP_6K0foI0,12399
+autogluon.core-1.2.1b20250301.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250301.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250301.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250301.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250301.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250301.dist-info/RECORD,,