autogluon.core 1.2.1b20250224__tar.gz → 1.2.1b20250226__tar.gz

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/PKG-INFO

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

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/src/autogluon/core/models/abstract/abstract_model.py

@@ -10,6 +10,7 @@ import pickle
 import sys
 import time
 from abc import ABC, abstractmethod
+from types import MappingProxyType
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import numpy as np
@@ -121,7 +122,7 @@ class Tunable(ABC):
         clone the current model.
         """
         pass
-        
+
     @abstractmethod
     def hyperparameter_tune(self, *args, **kwargs) -> tuple:
         pass
@@ -144,7 +145,7 @@ class ModelBase(Taggable, Tunable, ABC):
 
     @abstractmethod
     def get_info(self, *args, **kwargs) -> dict[str, Any]:
-        pass 
+        pass
 
     @abstractmethod
     def fit(self, *args, **kwargs) -> Self:
@@ -202,6 +203,10 @@ class AbstractModel(ModelBase):
         Hyperparameters that will be used by the model (can be search spaces instead of fixed values).
         If None, model defaults are used. This is identical to passing an empty dictionary.
     """
+    ag_key: str | None = None  # set to string value for subclasses for use in AutoGluon
+    ag_name: str | None = None  # set to string value for subclasses for use in AutoGluon
+    ag_priority: int = 0  # set to int value for subclasses for use in AutoGluon
+    ag_priority_by_problem_type: dict[str, int] = MappingProxyType({})  # if not set, we fall back to ag_priority. Use MappingProxyType to avoid mutation.
 
     model_file_name = "model.pkl"
     model_info_name = "info.pkl"
@@ -1018,10 +1023,7 @@ class AbstractModel(ModelBase):
         **kwargs :
             Any additional fit arguments a model supports.
         """
-        if "time_limit" in kwargs and kwargs["time_limit"] is not None:
-            time_start = time.time()
-        else:
-            time_start = None
+        time_start = time.time()
         kwargs = self.initialize(
             **kwargs
         )  # FIXME: This might have to go before self._preprocess_fit_args, but then time_limit might be incorrect in **kwargs init to initialize
@@ -2624,3 +2626,14 @@ class AbstractModel(ModelBase):
     def fit_num_gpus_child(self) -> float:
         """Number of GPUs used for fitting one model (i.e. a child model)"""
         return self.fit_num_gpus
+
+    @classmethod
+    def get_ag_priority(cls, problem_type: str | None = None) -> int:
+        """
+        Returns the AutoGluon fit priority,
+        defined by `cls.ag_priority` and `cls.ag_priority_by_problem_type`.
+        """
+        if problem_type is None:
+            return cls.ag_priority
+        else:
+            return cls.ag_priority_by_problem_type.get(problem_type, cls.ag_priority)

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/src/autogluon/core/models/dummy/dummy_model.py

@@ -10,6 +10,8 @@ class DummyModel(AbstractModel):
     A dummy model that ignores input features and predicts only a constant value.
     Useful for tests and calculating worst-case performance.
     """
+    ag_key = "DUMMY"
+    ag_name = "Dummy"
 
     def _get_model_type(self):
         from sklearn.dummy import DummyClassifier, DummyRegressor

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/src/autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py

@@ -10,6 +10,9 @@ logger = logging.getLogger(__name__)
 
 
 class GreedyWeightedEnsembleModel(AbstractModel):
+    ag_key = "ENS_WEIGHTED"
+    ag_name = "WeightedEnsemble"
+
     def __init__(self, base_model_names=None, model_base=EnsembleSelection, **kwargs):
         super().__init__(**kwargs)
         self.model_base = model_base
@@ -128,6 +131,9 @@ class GreedyWeightedEnsembleModel(AbstractModel):
 
 
 class SimpleWeightedEnsembleModel(GreedyWeightedEnsembleModel):
+    ag_key = "SIMPLE_ENS_WEIGHTED"
+    ag_name = "WeightedEnsemble"
+
     def __init__(self, model_base=SimpleWeightedEnsemble, **kwargs):
         super().__init__(model_base=model_base, **kwargs)
 

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/src/autogluon/core/version.py

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

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/src/autogluon.core.egg-info/PKG-INFO

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

{autogluon.core-1.2.1b20250224 → autogluon.core-1.2.1b20250226}/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.1b20250224
+autogluon.common==1.2.1b20250226
 
 [all]
 ray[default]<2.43,>=2.10.0
-hyperopt<0.2.8,>=0.2.7
 ray[default,tune]<2.43,>=2.10.0
 pyarrow>=15.0.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]
-flake8
-pytest
 types-requests
 pytest-mypy
+flake8
 types-setuptools
+pytest