autogluon.core 1.2.1b20250227__tar.gz → 1.2.1b20250301__tar.gz

{autogluon.core-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250227
+Version: 1.2.1b20250301
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.core-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/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-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/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-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/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-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/autogluon/core/version.py

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

{autogluon.core-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/autogluon.core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250227
+Version: 1.2.1b20250301
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.core-1.2.1b20250227 → autogluon.core-1.2.1b20250301}/src/autogluon.core.egg-info/requires.txt

@@ -7,13 +7,13 @@ tqdm<5,>=4.38
 requests
 matplotlib<3.11,>=3.7.0
 boto3<2,>=1.10
-autogluon.common==1.2.1b20250227
+autogluon.common==1.2.1b20250301
 
 [all]
+hyperopt<0.2.8,>=0.2.7
+ray[default]<2.43,>=2.10.0
 pyarrow>=15.0.0
 ray[default,tune]<2.43,>=2.10.0
-ray[default]<2.43,>=2.10.0
-hyperopt<0.2.8,>=0.2.7
 
 [ray]
 ray[default]<2.43,>=2.10.0
@@ -24,8 +24,8 @@ ray[default,tune]<2.43,>=2.10.0
 hyperopt<0.2.8,>=0.2.7
 
 [tests]
-types-requests
-pytest-mypy
 pytest
 flake8
+types-requests
+pytest-mypy
 types-setuptools