autogluon.tabular 1.4.1b20250923__tar.gz → 1.4.1b20251004__tar.gz

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

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

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

@@ -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.1b20250923 → autogluon.tabular-1.4.1b20251004}/src/autogluon/tabular/version.py

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

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

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

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