autogluon.tabular 1.4.1b20250921__tar.gz → 1.4.1b20250923__tar.gz

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/PKG-INFO

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

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/catboost/catboost_model.py

@@ -314,6 +314,8 @@ class CatBoostModel(AbstractModel):
         max_memory_iters = math.floor(available_mem * max_memory_proportion / mem_usage_per_iter)
 
         final_iters = min(default_iters, min(max_memory_iters, estimated_iters_in_time))
+        if final_iters < 1:
+            raise TimeLimitExceeded
         return final_iters
 
     def _predict_proba(self, X, **kwargs):

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/knn/knn_model.py

@@ -255,9 +255,13 @@ class KNNModel(AbstractModel):
             self._X_unused_index = [i for i in range(num_rows_max) if i not in idx]
         return self.model
 
-    def _get_maximum_resources(self) -> Dict[str, Union[int, float]]:
+    def _get_maximum_resources(self) -> dict[str, int | float]:
         # use at most 32 cpus to avoid OpenBLAS error: https://github.com/autogluon/autogluon/issues/1020
-        return {"num_cpus": 32}
+        # no GPU support
+        return {
+            "num_cpus": 32,
+            "num_gpus": 0,
+        }
 
     def _get_default_resources(self):
         # use at most 32 cpus to avoid OpenBLAS error: https://github.com/autogluon/autogluon/issues/1020

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/lgb/lgb_model.py

@@ -168,7 +168,7 @@ class LGBModel(AbstractModel):
                 #  Before enabling GPU, we should add code to detect that GPU-enabled version is installed and that a valid GPU exists.
                 #  GPU training heavily alters accuracy, often in a negative manner. We will have to be careful about when to use GPU.
                 params["device"] = "gpu"
-                logger.log(20, f"\tTraining {self.name} with GPU, note that this may negatively impact model quality compared to CPU training.")
+                logger.log(20, f"\tWarning: Training LightGBM with GPU. This may negatively impact model quality compared to CPU training.")
         logger.log(15, f"\tFitting {num_boost_round} rounds... Hyperparameters: {params}")
 
         if "num_threads" not in params:

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/lr/lr_model.py

@@ -322,6 +322,10 @@ class LinearModel(AbstractModel):
     ) -> int:
         return 4 * get_approximate_df_mem_usage(X).sum()
 
+    def _get_maximum_resources(self) -> dict[str, int | float]:
+        # no GPU support
+        return {"num_gpus": 0}
+
     @classmethod
     def supported_problem_types(cls) -> list[str] | None:
         return ["binary", "multiclass", "regression"]

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/rf/rf_model.py

@@ -370,6 +370,10 @@ class RFModel(AbstractModel):
 
         return self._convert_proba_to_unified_form(y_oof_pred_proba)
 
+    def _get_maximum_resources(self) -> dict[str, int | float]:
+        # no GPU support
+        return {"num_gpus": 0}
+
     def _get_default_auxiliary_params(self) -> dict:
         default_auxiliary_params = super()._get_default_auxiliary_params()
         extra_auxiliary_params = dict(

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/models/xgboost/xgboost_model.py

@@ -186,12 +186,18 @@ class XGBoostModel(AbstractModel):
         from xgboost import XGBClassifier, XGBRegressor
 
         model_type = XGBClassifier if self.problem_type in PROBLEM_TYPES_CLASSIFICATION else XGBRegressor
-        self.model = model_type(**params)
+
         import warnings
 
         with warnings.catch_warnings():
             # FIXME: v1.1: Upgrade XGBoost to 2.0.1+ to avoid deprecation warnings from Pandas 2.1+ during XGBoost fit.
             warnings.simplefilter(action="ignore", category=FutureWarning)
+            if params.get("device", "cpu") == "cuda:0":
+                # verbosity=0 to hide UserWarning: Falling back to prediction using DMatrix due to mismatched devices.
+                # TODO: Find a way to hide this warning without setting verbosity=0
+                #  ref: https://github.com/dmlc/xgboost/issues/9791
+                params["verbosity"] = 0
+            self.model = model_type(**params)
             self.model.fit(X=X, y=y, eval_set=eval_set, verbose=False, sample_weight=sample_weight)
 
         if generate_curves:

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.4.1b20250921"
+__version__ = "1.4.1b20250923"
 __lite__ = False

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.4.1b20250921 → autogluon.tabular-1.4.1b20250923}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,24 +3,24 @@ scipy<1.17,>=1.5.4
 pandas<2.4.0,>=2.0.0
 scikit-learn<1.8.0,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.4.1b20250921
-autogluon.features==1.4.1b20250921
+autogluon.core==1.4.1b20250923
+autogluon.features==1.4.1b20250923
 
 [all]
 loguru
-einx
-omegaconf
-catboost<1.3,>=1.2
-einops<0.9,>=0.7
+autogluon.core[all]==1.4.1b20250923
 fastai<2.9,>=2.3.1
-numpy<2.3.0,>=1.25
+torch<2.8,>=2.6
 lightgbm<4.7,>=4.0
-autogluon.core[all]==1.4.1b20250921
+einx
 transformers
-spacy<3.9
-xgboost<3.1,>=2.0
-torch<2.8,>=2.6
 huggingface_hub[torch]
+einops<0.9,>=0.7
+xgboost<3.1,>=2.0
+omegaconf
+numpy<2.3.0,>=1.25
+catboost<1.3,>=1.2
+spacy<3.9
 
 [all:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0
@@ -56,7 +56,7 @@ huggingface_hub[torch]
 einops<0.9,>=0.7
 
 [ray]
-autogluon.core[all]==1.4.1b20250921
+autogluon.core[all]==1.4.1b20250923
 
 [realmlp]
 pytabkit<1.7,>=1.6
@@ -76,24 +76,24 @@ onnx<1.18.0,>=1.13.0
 onnx<1.16.2,>=1.13.0
 
 [tabarena]
-einops<0.9,>=0.7
 tabicl<0.2,>=0.1.3
-pytabkit<1.7,>=1.6
-torch<2.8,>=2.6
+autogluon.core[all]==1.4.1b20250923
+lightgbm<4.7,>=4.0
 einx
-fastai<2.9,>=2.3.1
+einops<0.9,>=0.7
 numpy<2.3.0,>=1.25
-spacy<3.9
-omegaconf
 catboost<1.3,>=1.2
+loguru
+torch<2.8,>=2.6
 interpret-core<0.8,>=0.7.2
-lightgbm<4.7,>=4.0
-transformers
+spacy<3.9
+huggingface_hub[torch]
+pytabkit<1.7,>=1.6
 tabpfn<2.2,>=2.0.9
+omegaconf
+fastai<2.9,>=2.3.1
+transformers
 xgboost<3.1,>=2.0
-loguru
-autogluon.core[all]==1.4.1b20250921
-huggingface_hub[torch]
 
 [tabarena:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0