autogluon.tabular 1.3.1b20250527__tar.gz → 1.3.2b20250528__tar.gz

{autogluon.tabular-1.3.1b20250527 → autogluon.tabular-1.3.2b20250528}/PKG-INFO

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

{autogluon.tabular-1.3.1b20250527 → autogluon.tabular-1.3.2b20250528}/src/autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py

@@ -8,7 +8,7 @@ import random
 import time
 import warnings
 from copy import deepcopy
-from typing import Dict, Union, TYPE_CHECKING
+from typing import TYPE_CHECKING, Dict, Union
 
 import numpy as np
 import pandas as pd
@@ -103,9 +103,14 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
         if num_gpus is not None and num_gpus >= 1:
             if torch.cuda.is_available():
                 device = torch.device("cuda")
-                logger.log(15, "Training on GPU")
+                logger.log(15, "Training on GPU (CUDA)")
                 if num_gpus > 1:
                     logger.warning(f"{self.__class__.__name__} not yet able to use more than 1 GPU. 'num_gpus' is set to >1, but we will be using only 1 GPU.")
+            elif hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
+                device = torch.device("mps")
+                logger.log(15, "Training on GPU (MPS - Apple Silicon)")
+                if num_gpus > 1:
+                    logger.warning(f"{self.__class__.__name__} on Apple Silicon can only use 1 GPU (MPS). 'num_gpus' is set to >1, but we will be using only 1 GPU.")
             else:
                 device = torch.device("cpu")
                 logger.log(15, "Training on CPU")

{autogluon.tabular-1.3.1b20250527 → autogluon.tabular-1.3.2b20250528}/src/autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.1b20250527"
+__version__ = "1.3.2b20250528"
 __lite__ = False

{autogluon.tabular-1.3.1b20250527 → autogluon.tabular-1.3.2b20250528}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.3.1b20250527 → autogluon.tabular-1.3.2b20250528}/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.7.0,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.3.1b20250527
-autogluon.features==1.3.1b20250527
+autogluon.core==1.3.2b20250528
+autogluon.features==1.3.2b20250528
 
 [all]
-catboost<1.3,>=1.2
-xgboost<3.1,>=2.0
+einops<0.9,>=0.7
+numpy<2.3.0,>=1.25
 lightgbm<4.7,>=4.0
+huggingface_hub[torch]
 fastai<2.9,>=2.3.1
-spacy<3.9
 torch<2.7,>=2.2
-autogluon.core[all]==1.3.1b20250527
-numpy<2.3.0,>=1.25
-einops<0.9,>=0.7
-huggingface_hub[torch]
+autogluon.core[all]==1.3.2b20250528
+catboost<1.3,>=1.2
+spacy<3.9
+xgboost<3.1,>=2.0
 
 [catboost]
 numpy<2.3.0,>=1.25
@@ -34,7 +34,7 @@ imodels<2.1.0,>=1.3.10
 lightgbm<4.7,>=4.0
 
 [ray]
-autogluon.core[all]==1.3.1b20250527
+autogluon.core[all]==1.3.2b20250528
 
 [skex]
 scikit-learn-intelex<2025.5,>=2024.0