autogluon.tabular 1.2.1b20250228__tar.gz → 1.2.1b20250302__tar.gz

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/PKG-INFO

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

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/learner/abstract_learner.py

@@ -235,6 +235,9 @@ class AbstractTabularLearner(AbstractLearner):
                 y_pred = y_pred.values
         else:
             if as_pandas:
+                if len(y_pred) == 0:
+                    # avoid exception due to mismatched shape for empty predict
+                    y_pred = None
                 y_pred = pd.DataFrame(data=y_pred, columns=self.quantile_levels, index=index)
         return y_pred
 

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/catboost/catboost_model.py

@@ -344,6 +344,10 @@ class CatBoostModel(AbstractModel):
         num_gpus = 0
         return num_cpus, num_gpus
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile", "softclass"]
+
     @classmethod
     def _class_tags(cls):
         return {

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/fastainn/tabular_nn_fastai.py

@@ -649,6 +649,10 @@ class NNFastAiTabularModel(AbstractModel):
             minimum_resources["num_gpus"] = 0.5
         return minimum_resources
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile"]
+
     @classmethod
     def _class_tags(cls):
         return {"can_estimate_memory_usage_static": True}

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/knn/knn_model.py

@@ -74,7 +74,6 @@ class KNNModel(AbstractModel):
         default_ag_args = super()._get_default_ag_args()
         extra_ag_args = {
             "valid_stacker": False,
-            "problem_types": [BINARY, MULTICLASS, REGRESSION],
         }
         default_ag_args.update(extra_ag_args)
         return default_ag_args
@@ -264,6 +263,10 @@ class KNNModel(AbstractModel):
         num_gpus = 0
         return num_cpus, num_gpus
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression"]
+
     @classmethod
     def _class_tags(cls):
         return {

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/lgb/lgb_model.py

@@ -528,6 +528,10 @@ class LGBModel(AbstractModel):
         num_gpus = 0
         return num_cpus, num_gpus
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile", "softclass"]
+
     @property
     def _features(self):
         return self._features_internal_list

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/lr/lr_model.py

@@ -317,6 +317,10 @@ class LinearModel(AbstractModel):
     ) -> int:
         return 4 * get_approximate_df_mem_usage(X).sum()
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression"]
+
     @classmethod
     def _class_tags(cls):
         return {"can_estimate_memory_usage_static": True}

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/rf/rf_model.py

@@ -381,6 +381,10 @@ class RFModel(AbstractModel):
             default_ag_args_ensemble.update(extra_ag_args_ensemble)
         return default_ag_args_ensemble
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile", "softclass"]
+
     @classmethod
     def _class_tags(cls):
         return {"can_estimate_memory_usage_static": True}

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py

@@ -946,6 +946,10 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
         )
         self.processor = self._compiler.compile(model=(self.processor, self.model), path=self.path, input_types=input_types)
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile", "softclass"]
+
     @classmethod
     def _class_tags(cls):
         return {

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/xgboost/xgboost_model.py

@@ -10,7 +10,7 @@ from autogluon.common.utils.lite import disable_if_lite_mode
 from autogluon.common.utils.pandas_utils import get_approximate_df_mem_usage
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.common.utils.try_import import try_import_xgboost
-from autogluon.core.constants import BINARY, MULTICLASS, PROBLEM_TYPES_CLASSIFICATION, REGRESSION, SOFTCLASS
+from autogluon.core.constants import MULTICLASS, PROBLEM_TYPES_CLASSIFICATION, REGRESSION, SOFTCLASS
 from autogluon.core.models import AbstractModel
 from autogluon.core.models._utils import get_early_stopping_rounds
 
@@ -45,15 +45,6 @@ class XGBoostModel(AbstractModel):
     def _get_default_searchspace(self):
         return get_default_searchspace(problem_type=self.problem_type, num_classes=self.num_classes)
 
-    @classmethod
-    def _get_default_ag_args(cls) -> dict:
-        default_ag_args = super()._get_default_ag_args()
-        extra_ag_args = {
-            "problem_types": [BINARY, MULTICLASS, REGRESSION, SOFTCLASS],
-        }
-        default_ag_args.update(extra_ag_args)
-        return default_ag_args
-
     def _get_default_auxiliary_params(self) -> dict:
         default_auxiliary_params = super()._get_default_auxiliary_params()
         extra_auxiliary_params = dict(
@@ -325,6 +316,10 @@ class XGBoostModel(AbstractModel):
             model._xgb_model_type = None
         return model
 
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "softclass"]
+
     @classmethod
     def _class_tags(cls):
         return {

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/models/xt/xt_model.py

@@ -24,3 +24,7 @@ class XTModel(RFModel):
             from sklearn.ensemble import ExtraTreesClassifier
 
             return ExtraTreesClassifier
+
+    @classmethod
+    def supported_problem_types(cls) -> list[str] | None:
+        return ["binary", "multiclass", "regression", "quantile"]

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon/tabular/version.py

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

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.2.1b20250228 → autogluon.tabular-1.2.1b20250302}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,20 +3,20 @@ scipy<1.16,>=1.5.4
 pandas<2.3.0,>=2.0.0
 scikit-learn<1.5.3,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.2.1b20250228
-autogluon.features==1.2.1b20250228
+autogluon.core==1.2.1b20250302
+autogluon.features==1.2.1b20250302
 
 [all]
+lightgbm<4.6,>=4.0
 huggingface_hub[torch]
-torch<2.6,>=2.2
+spacy<3.8
 fastai<2.8,>=2.3.1
 einops<0.9,>=0.7
-autogluon.core[all]==1.2.1b20250228
-numpy<2.0.0,>=1.25
+autogluon.core[all]==1.2.1b20250302
 xgboost<2.2,>=1.6
 catboost<1.3,>=1.2
-spacy<3.8
-lightgbm<4.6,>=4.0
+numpy<2.0.0,>=1.25
+torch<2.6,>=2.2
 
 [catboost]
 numpy<2.0.0,>=1.25
@@ -34,7 +34,7 @@ imodels<1.4.0,>=1.3.10
 lightgbm<4.6,>=4.0
 
 [ray]
-autogluon.core[all]==1.2.1b20250228
+autogluon.core[all]==1.2.1b20250302
 
 [skex]
 scikit-learn-intelex<2025.1,>=2024.0