autogluon.tabular 1.3.2b20250718__py3-none-any.whl → 1.3.2b20250719__py3-none-any.whl

autogluon/tabular/models/mitra/mitra_model.py

@@ -1,18 +1,23 @@
-# TODO: To ensure deterministic operations we need to set torch.use_deterministic_algorithms(True) 
+# TODO: To ensure deterministic operations we need to set torch.use_deterministic_algorithms(True)
 # and os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'. The CUBLAS environment variable configures
 # the workspace size for certain CUBLAS operations to ensure reproducibility when using CUDA >= 10.2.
 # Both settings are required to ensure deterministic behavior in operations such as matrix multiplications.
 import os
-os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
+
+os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
 
 import os
 from typing import List, Optional
 
 import pandas as pd
+import torch
+import logging
 
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.core.models import AbstractModel
 
+logger = logging.getLogger(__name__)
+
 
 # TODO: Needs memory usage estimate method
 class MitraModel(AbstractModel):
@@ -26,12 +31,26 @@ class MitraModel(AbstractModel):
         self.problem_type = problem_type
         self._weights_saved = False
 
+    @staticmethod
+    def _get_default_device():
+        """Get the best available device for the current system."""
+        if ResourceManager.get_gpu_count_torch(cuda_only=True) > 0:
+            logger.info("Using CUDA GPU")
+            return "cuda"
+        elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
+            logger.info("Using MPS GPU")
+            return "mps"  # Apple silicon
+        else:
+            return "cpu"
+
     def get_model_cls(self):
         from .sklearn_interface import MitraClassifier
-        if self.problem_type in ['binary', 'multiclass']:
+
+        if self.problem_type in ["binary", "multiclass"]:
             model_cls = MitraClassifier
-        elif self.problem_type == 'regression':
+        elif self.problem_type == "regression":
             from .sklearn_interface import MitraRegressor
+
             model_cls = MitraRegressor
         else:
             raise AssertionError(f"Unsupported problem_type: {self.problem_type}")
@@ -47,7 +66,6 @@ class MitraModel(AbstractModel):
         num_cpus: int = 1,
         **kwargs,
     ):
-        
         # TODO: Reset the number of threads based on the specified num_cpus
         need_to_reset_torch_threads = False
         torch_threads_og = None
@@ -91,7 +109,7 @@ class MitraModel(AbstractModel):
 
     def _set_default_params(self):
         default_params = {
-            "device": "cpu",
+            "device": self._get_default_device(),
             "n_estimators": 1,
         }
         for param, val in default_params.items():
@@ -127,6 +145,7 @@ class MitraModel(AbstractModel):
         path = super().save(path=path, verbose=verbose)
         if _model_weights_list is not None:
             import torch
+
             os.makedirs(self.path, exist_ok=True)
             torch.save(_model_weights_list, self.weights_path)
             for i in range(len(self.model.trainers)):
@@ -139,6 +158,7 @@ class MitraModel(AbstractModel):
 
         if model._weights_saved:
             import torch
+
             model_weights_list = torch.load(model.weights_path, weights_only=False)  # nosec B614
             for i in range(len(model.model.trainers)):
                 model.model.trainers[i].model = model_weights_list[i]
@@ -154,7 +174,7 @@ class MitraModel(AbstractModel):
         default_ag_args_ensemble = super()._get_default_ag_args_ensemble(**kwargs)
         # FIXME: Test if it works with parallel, need to enable n_cpus support
         extra_ag_args_ensemble = {
-           "fold_fitting_strategy": "sequential_local",  # FIXME: Comment out after debugging for large speedup
+            "fold_fitting_strategy": "sequential_local",  # FIXME: Comment out after debugging for large speedup
         }
         default_ag_args_ensemble.update(extra_ag_args_ensemble)
         return default_ag_args_ensemble
@@ -168,7 +188,9 @@ class MitraModel(AbstractModel):
         return num_cpus, num_gpus
 
     def _estimate_memory_usage(self, X: pd.DataFrame, **kwargs) -> int:
-        return self.estimate_memory_usage_static(X=X, problem_type=self.problem_type, num_classes=self.num_classes, **kwargs)
+        return self.estimate_memory_usage_static(
+            X=X, problem_type=self.problem_type, num_classes=self.num_classes, **kwargs
+        )
 
     @classmethod
     def _estimate_memory_usage_static(
@@ -199,10 +221,9 @@ class MitraModel(AbstractModel):
             cpu_memory_kb = 1.3 * (100 * rows * features + 1000000)  # 1GB base + linear scaling
         else:
             # Original formula for larger datasets
-            cpu_memory_kb = 1.3 * (0.001748 * (rows**2) * features + \
-                            0.001206 * rows * (features**2) + \
-                            10.3482 * rows * features + \
-                            6409698)
+            cpu_memory_kb = 1.3 * (
+                0.001748 * (rows**2) * features + 0.001206 * rows * (features**2) + 10.3482 * rows * features + 6409698
+            )
         return int(cpu_memory_kb * 1e3)
 
     @classmethod
@@ -220,10 +241,9 @@ class MitraModel(AbstractModel):
             cpu_memory_kb = 1.3 * (200 * rows * features + 2000000)  # 2GB base + linear scaling
         else:
             # Original formula for larger datasets
-            cpu_memory_kb = 1.3 * (0.001 * (rows**2) * features + \
-                            0.004541 * rows * (features**2) + \
-                            46.2974 * rows * features + \
-                            5605681)
+            cpu_memory_kb = 1.3 * (
+                0.001 * (rows**2) * features + 0.004541 * rows * (features**2) + 46.2974 * rows * features + 5605681
+            )
         return int(cpu_memory_kb * 1e3)
 
     @classmethod

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250718"
+__version__ = "1.3.2b20250719"
 __lite__ = False

{autogluon.tabular-1.3.2b20250718.dist-info → autogluon.tabular-1.3.2b20250719.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250718
+Version: 1.3.2b20250719
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,20 +41,20 @@ Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: scikit-learn<1.8.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.3.2b20250718
-Requires-Dist: autogluon.features==1.3.2b20250718
+Requires-Dist: autogluon.core==1.3.2b20250719
+Requires-Dist: autogluon.features==1.3.2b20250719
 Provides-Extra: all
+Requires-Dist: autogluon.core[all]==1.3.2b20250719; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: pytabkit<1.6,>=1.5; extra == "all"
 Requires-Dist: torch<2.8,>=2.2; extra == "all"
-Requires-Dist: autogluon.core[all]==1.3.2b20250718; extra == "all"
 Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
-Requires-Dist: pytabkit<1.6,>=1.5; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -72,7 +72,7 @@ Requires-Dist: einx; extra == "mitra"
 Requires-Dist: omegaconf; extra == "mitra"
 Requires-Dist: transformers; extra == "mitra"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.3.2b20250718; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.3.2b20250719; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "realmlp"
 Provides-Extra: skex

{autogluon.tabular-1.3.2b20250718.dist-info → autogluon.tabular-1.3.2b20250719.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.3.2b20250718-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.3.2b20250719-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=t7hPQFF0BzYTBfD-vM9hoER3q-C5x0pjSWoVO1dcT0w,91
+autogluon/tabular/version.py,sha256=YLKHqDF99tyKxzCGdB6cvf6IZT4wW9FS65QYC0TWIYM,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=JsdVGmpcYL88GPKBznPtqJ1sGaByOSvLn7KWU-HyVoQ,21085
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -68,7 +68,7 @@ autogluon/tabular/models/lr/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TIm
 autogluon/tabular/models/lr/hyperparameters/parameters.py,sha256=Hr5YC13zjbt3CfCbzGj8iXUIuDn-Q7FvDT2uSuiSVlM,1414
 autogluon/tabular/models/lr/hyperparameters/searchspaces.py,sha256=Igywc-B6qJ9EBLdasrDhW-Ot5FGirIzbXLwv5HRe5Xo,276
 autogluon/tabular/models/mitra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/mitra/mitra_model.py,sha256=XiTlzy-RbbHe1t8VCU1y976zmqrDAkO_HhkBiSlk7mM,9985
+autogluon/tabular/models/mitra/mitra_model.py,sha256=gvoh22xk17G9ctcYL3RDDWkFcqEbFsJsNKXClPHSK8M,10429
 autogluon/tabular/models/mitra/sklearn_interface.py,sha256=nX830-_7KYjMnwJ8m8jhCfG7BXU379Ecn5Lu3RvN8Us,18513
 autogluon/tabular/models/mitra/_internal/__init__.py,sha256=dN2dz1pGMgQTFiSf9oYbyq23iJUxV8QNlOX3qw3KUO4,35
 autogluon/tabular/models/mitra/_internal/config/__init__.py,sha256=Exu_Sx6-K-D5peDQ_TibsjZpqAALs2-9IXfq8hu1mwU,40
@@ -188,11 +188,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=hoWADaOG576Q_XLV1nY_ju
 autogluon/tabular/trainer/model_presets/presets_distill.py,sha256=MnFC2GJc6RmDBNAGbsO2XMfo3PjR8cUrZoilWW8gTYQ,3295
 autogluon/tabular/tuning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/tuning/feature_pruner.py,sha256=9iNku8gVbYEkjuKlyITPJDicsNkoraaQOlINQq9iZlQ,6877
-autogluon.tabular-1.3.2b20250718.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.3.2b20250718.dist-info/METADATA,sha256=edzh0r-bATMf-HfQBg7-Gdox3Bq8KfuT101utiZe35s,14646
-autogluon.tabular-1.3.2b20250718.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.3.2b20250718.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.3.2b20250718.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250718.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250718.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.3.2b20250718.dist-info/RECORD,,
+autogluon.tabular-1.3.2b20250719.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.3.2b20250719.dist-info/METADATA,sha256=KRUVAwkQN8aBxfrNHAd02IIgTbtJx-cxqm9rfUP4E68,14646
+autogluon.tabular-1.3.2b20250719.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.3.2b20250719.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.3.2b20250719.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250719.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250719.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.3.2b20250719.dist-info/RECORD,,