autogluon.tabular 1.4.1b20250916__tar.gz → 1.4.1b20250929__tar.gz

{autogluon.tabular-1.4.1b20250916 → autogluon.tabular-1.4.1b20250929}/PKG-INFO

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

{autogluon.tabular-1.4.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916 → autogluon.tabular-1.4.1b20250929}/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:
@@ -305,16 +305,28 @@ class LGBModel(AbstractModel):
             try:
                 self.model = train_lgb_model(early_stopping_callback_kwargs=early_stopping_callback_kwargs, **train_params)
             except LightGBMError:
-                if train_params["params"].get("device", "cpu") != "gpu":
+                if train_params["params"].get("device", "cpu") not in ["gpu", "cuda"]:
                     raise
                 else:
-                    logger.warning(
-                        "Warning: GPU mode might not be installed for LightGBM, GPU training raised an exception. Falling back to CPU training..."
-                        "Refer to LightGBM GPU documentation: https://github.com/Microsoft/LightGBM/tree/master/python-package#build-gpu-version"
-                        "One possible method is:"
-                        "\tpip uninstall lightgbm -y"
-                        "\tpip install lightgbm --install-option=--gpu"
-                    )
+                    if train_params["params"]["device"] == "gpu":
+                        logger.warning(
+                            "Warning: GPU mode might not be installed for LightGBM, "
+                            "GPU training raised an exception. Falling back to CPU training..."
+                            "Refer to LightGBM GPU documentation: "
+                            "https://github.com/Microsoft/LightGBM/tree/master/python-package#build-gpu-version"
+                            "One possible method is:"
+                            "\tpip uninstall lightgbm -y"
+                            "\tpip install lightgbm --install-option=--gpu"
+                        )
+                    elif train_params["params"]["device"] == "cuda":
+                        # Current blocker for using CUDA over GPU: https://github.com/microsoft/LightGBM/issues/6828
+                        # Note that device="cuda" works if AutoGluon (and therefore LightGBM) is installed via conda.
+                        logger.warning(
+                            "Warning: CUDA mode might not be installed for LightGBM, "
+                            "CUDA training raised an exception. Falling back to CPU training..."
+                            "Refer to LightGBM CUDA documentation: "
+                            "https://github.com/Microsoft/LightGBM/tree/master/python-package#build-cuda-version"
+                        )
                     train_params["params"]["device"] = "cpu"
                     self.model = train_lgb_model(early_stopping_callback_kwargs=early_stopping_callback_kwargs, **train_params)
             retrain = False
@@ -515,17 +527,44 @@ class LGBModel(AbstractModel):
         default_auxiliary_params.update(extra_auxiliary_params)
         return default_auxiliary_params
 
-    def _is_gpu_lgbm_installed(self):
+    @staticmethod
+    def _is_gpu_lgbm_installed():
         # Taken from https://github.com/microsoft/LightGBM/issues/3939
         try_import_lightgbm()
         import lightgbm
 
+        rng = np.random.RandomState(42)
+        data = rng.rand(25, 2)
+        label = rng.randint(2, size=25)
+
+        try:
+            train_data = lightgbm.Dataset(data, label=label)
+            params = {
+                "device": "gpu",
+                "verbose": -1,
+            }
+            gbm = lightgbm.train(params, num_boost_round=10, train_set=train_data)
+            return True
+        except Exception as e:
+            return False
+
+    @staticmethod
+    def _is_cuda_lgbm_installed():
+        # Taken from https://github.com/microsoft/LightGBM/issues/3939
+        try_import_lightgbm()
+        import lightgbm
+
+        rng = np.random.RandomState(42)
+        data = rng.rand(25, 2)
+        label = rng.randint(2, size=25)
+
         try:
-            data = np.random.rand(50, 2)
-            label = np.random.randint(2, size=50)
             train_data = lightgbm.Dataset(data, label=label)
-            params = {"device": "gpu"}
-            gbm = lightgbm.train(params, train_set=train_data, verbose=-1)
+            params = {
+                "device": "cuda",
+                "verbose": -1,
+            }
+            gbm = lightgbm.train(params, num_boost_round=10, train_set=train_data)
             return True
         except Exception as e:
             return False
@@ -534,7 +573,7 @@ class LGBModel(AbstractModel):
         minimum_resources = {
             "num_cpus": 1,
         }
-        if is_gpu_available and self._is_gpu_lgbm_installed():
+        if is_gpu_available:
             minimum_resources["num_gpus"] = 0.5
         return minimum_resources
 

{autogluon.tabular-1.4.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916 → autogluon.tabular-1.4.1b20250929}/src/autogluon/tabular/version.py

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

{autogluon.tabular-1.4.1b20250916 → autogluon.tabular-1.4.1b20250929}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.4.1b20250916 → autogluon.tabular-1.4.1b20250929}/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.1b20250916
-autogluon.features==1.4.1b20250916
+autogluon.core==1.4.1b20250929
+autogluon.features==1.4.1b20250929
 
 [all]
-fastai<2.9,>=2.3.1
+einops<0.9,>=0.7
 lightgbm<4.7,>=4.0
-spacy<3.9
-transformers
-loguru
-torch<2.8,>=2.6
-numpy<2.3.0,>=1.25
-omegaconf
-autogluon.core[all]==1.4.1b20250916
 catboost<1.3,>=1.2
-einops<0.9,>=0.7
 xgboost<3.1,>=2.0
+transformers
+spacy<3.9
+torch<2.8,>=2.6
+fastai<2.9,>=2.3.1
+loguru
 einx
+omegaconf
+autogluon.core[all]==1.4.1b20250929
 huggingface_hub[torch]
+numpy<2.3.0,>=1.25
 
 [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.1b20250916
+autogluon.core[all]==1.4.1b20250929
 
 [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]
+tabicl<0.2,>=0.1.3
+omegaconf
 fastai<2.9,>=2.3.1
-torch<2.8,>=2.6
-pytabkit<1.7,>=1.6
-einops<0.9,>=0.7
-einx
-transformers
+spacy<3.9
 tabpfn<2.2,>=2.0.9
+numpy<2.3.0,>=1.25
+lightgbm<4.7,>=4.0
+transformers
+torch<2.8,>=2.6
+autogluon.core[all]==1.4.1b20250929
 loguru
 interpret-core<0.8,>=0.7.2
-omegaconf
-autogluon.core[all]==1.4.1b20250916
-lightgbm<4.7,>=4.0
+pytabkit<1.7,>=1.6
+einops<0.9,>=0.7
 catboost<1.3,>=1.2
 xgboost<3.1,>=2.0
+einx
 huggingface_hub[torch]
-tabicl<0.2,>=0.1.3
-spacy<3.9
-numpy<2.3.0,>=1.25
 
 [tabarena:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0