autogluon.tabular 1.3.2b20250721__py3-none-any.whl → 1.3.2b20250722__py3-none-any.whl

autogluon/tabular/models/mitra/mitra_model.py

@@ -37,9 +37,6 @@ class MitraModel(AbstractModel):
         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"
 

autogluon/tabular/models/mitra/sklearn_interface.py

@@ -12,6 +12,7 @@ from ._internal.config.enums import ModelName
 from ._internal.core.trainer_finetune import TrainerFinetune
 from ._internal.data.dataset_split import make_stratified_dataset_split
 from ._internal.models.tab2d import Tab2D
+from ._internal.utils.set_seed import set_seed
 
 # Hyperparameter search space
 DEFAULT_FINE_TUNE = True # [True, False]
@@ -114,6 +115,7 @@ class MitraBase(BaseEstimator):
         self.train_time = 0
         self.seed = seed
 
+        set_seed(self.seed)
 
     def _create_config(self, task, dim_output, time_limit=None):
         cfg = ConfigRun(

autogluon/tabular/version.py

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

{autogluon.tabular-1.3.2b20250721.dist-info → autogluon.tabular-1.3.2b20250722.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250721
+Version: 1.3.2b20250722
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,20 +41,21 @@ 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.2b20250721
-Requires-Dist: autogluon.features==1.3.2b20250721
+Requires-Dist: autogluon.core==1.3.2b20250722
+Requires-Dist: autogluon.features==1.3.2b20250722
 Provides-Extra: all
-Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: torch<2.8,>=2.2; extra == "all"
-Requires-Dist: autogluon.core[all]==1.3.2b20250721; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: huggingface-hub[torch]; extra == "all"
 Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
 Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
+Requires-Dist: autogluon.core[all]==1.3.2b20250722; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: torch<2.8,>=2.2; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -62,6 +63,7 @@ Provides-Extra: fastai
 Requires-Dist: spacy<3.9; extra == "fastai"
 Requires-Dist: torch<2.8,>=2.2; extra == "fastai"
 Requires-Dist: fastai<2.9,>=2.3.1; extra == "fastai"
+Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "fastai"
 Provides-Extra: imodels
 Requires-Dist: imodels<2.1.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
@@ -72,7 +74,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.2b20250721; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.3.2b20250722; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "realmlp"
 Provides-Extra: skex

{autogluon.tabular-1.3.2b20250721.dist-info → autogluon.tabular-1.3.2b20250722.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.3.2b20250721-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.3.2b20250722-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=vXUGGCs1QRFJHTMsLjx-7G1L4whzBB-L9cIVuXLM7hI,91
+autogluon/tabular/version.py,sha256=CdIaQfJPPjnE9mEFR_3W2YJdiCuNqvKW-cgu00I6i3w,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,8 +68,8 @@ 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=gvoh22xk17G9ctcYL3RDDWkFcqEbFsJsNKXClPHSK8M,10429
-autogluon/tabular/models/mitra/sklearn_interface.py,sha256=nX830-_7KYjMnwJ8m8jhCfG7BXU379Ecn5Lu3RvN8Us,18513
+autogluon/tabular/models/mitra/mitra_model.py,sha256=SQehZMs9WBP9o51CJ6L8njuEq0UcunXbbOM3eedhokw,10263
+autogluon/tabular/models/mitra/sklearn_interface.py,sha256=p3keXfA3cdQhoT24jAytrC1uHOK8tQXpE2tSWagAQzU,18588
 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
 autogluon/tabular/models/mitra/_internal/config/config_pretrain.py,sha256=CeaD96EcDX69LdcLTYGlFmYLdBNINEJXRMWmJ6LbhTg,6038
@@ -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.2b20250721.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.3.2b20250721.dist-info/METADATA,sha256=oqWixsqkCeZZz1kPOldPmhIsw08dxJ4C9b-BkKm7UcU,14646
-autogluon.tabular-1.3.2b20250721.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.3.2b20250721.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.3.2b20250721.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250721.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250721.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.3.2b20250721.dist-info/RECORD,,
+autogluon.tabular-1.3.2b20250722.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.3.2b20250722.dist-info/METADATA,sha256=wtSNYXyA4ratSdv6547JHxGPrJOyU8Xq-8j8hFhZcho,14875
+autogluon.tabular-1.3.2b20250722.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.3.2b20250722.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.3.2b20250722.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250722.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250722.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.3.2b20250722.dist-info/RECORD,,