autogluon.core 1.2.1b20250305__tar.gz → 1.2.1b20250306__tar.gz

{autogluon.core-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/PKG-INFO

@@ -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

{autogluon.core-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/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-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/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-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/autogluon/core/models/ensemble/weighted_ensemble_model.py

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

{autogluon.core-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/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-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/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 → autogluon.core-1.2.1b20250306}/src/autogluon.core.egg-info/PKG-INFO

@@ -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

{autogluon.core-1.2.1b20250305 → autogluon.core-1.2.1b20250306}/src/autogluon.core.egg-info/requires.txt

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