autogluon.tabular 1.4.0b20250725__py3-none-any.whl → 1.4.0b20250726__py3-none-any.whl

autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2025.py

@@ -303,6 +303,7 @@ hyperparameter_portfolio_zeroshot_2025_small = {
             "n_estimators": 1,
             "fine_tune": True,
             "fine_tune_steps": 50,
+            "ag.num_gpus": 1,
             "ag_args": {'priority': -21},
         },
     ],

autogluon/tabular/models/mitra/mitra_model.py

@@ -119,6 +119,13 @@ class MitraModel(AbstractModel):
             else:
                 hyp["device"] = self._get_default_device()
 
+        if hyp["device"] == "cpu" and hyp.get("fine_tune", True):
+            logger.log(
+                30,
+                f"\tWarning: Attempting to fine-tune Mitra on CPU. This will be very slow. "
+                f"We strongly recommend using a GPU instance to fine-tune Mitra."
+            )
+
         if "state_dict_classification" in hyp:
             state_dict_classification = hyp.pop("state_dict_classification")
             if self.problem_type in ["binary", "multiclass"]:
@@ -229,24 +236,6 @@ class MitraModel(AbstractModel):
 
         return num_cpus, num_gpus
 
-    def get_minimum_resources(self, is_gpu_available: bool = False) -> dict[str, int | float]:
-        """
-        Parameters
-        ----------
-        is_gpu_available : bool, default = False
-            Whether gpu is available in the system.
-            Model that can be trained both on cpu and gpu can decide the minimum resources based on this.
-
-        Returns a dictionary of minimum resource requirements to fit the model.
-        Subclass should consider overriding this method if it requires more resources to train.
-        If a resource is not part of the output dictionary, it is considered unnecessary.
-        Valid keys: 'num_cpus', 'num_gpus'.
-        """
-        return {
-            "num_cpus": 1,
-            "num_gpus": 0.5,
-        }
-
     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

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.4.0b20250725"
+__version__ = "1.4.0b20250726"
 __lite__ = False

{autogluon.tabular-1.4.0b20250725.dist-info → autogluon.tabular-1.4.0b20250726.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.4.0b20250725
+Version: 1.4.0b20250726
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,22 +41,22 @@ 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.4.0b20250725
-Requires-Dist: autogluon.features==1.4.0b20250725
+Requires-Dist: autogluon.core==1.4.0b20250726
+Requires-Dist: autogluon.features==1.4.0b20250726
 Provides-Extra: all
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
 Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: torch<2.8,>=2.2; extra == "all"
-Requires-Dist: einx; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Requires-Dist: loguru; extra == "all"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: transformers; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: einx; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: autogluon.core[all]==1.4.0b20250725; extra == "all"
 Requires-Dist: omegaconf; extra == "all"
-Requires-Dist: transformers; extra == "all"
+Requires-Dist: autogluon.core[all]==1.4.0b20250726; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; 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"
@@ -78,7 +78,7 @@ Requires-Dist: torch<2.8,>=2.2; extra == "mitra"
 Requires-Dist: transformers; extra == "mitra"
 Requires-Dist: huggingface-hub[torch]; extra == "mitra"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.4.0b20250725; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.4.0b20250726; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "realmlp"
 Provides-Extra: skex
@@ -90,22 +90,22 @@ Requires-Dist: onnxruntime-gpu<1.20.0,>=1.17.0; extra == "skl2onnx"
 Requires-Dist: onnx<1.18.0,>=1.13.0; platform_system != "Windows" and extra == "skl2onnx"
 Requires-Dist: onnx<1.16.2,>=1.13.0; platform_system == "Windows" and extra == "skl2onnx"
 Provides-Extra: tabarena
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
 Requires-Dist: xgboost<3.1,>=2.0; extra == "tabarena"
 Requires-Dist: torch<2.8,>=2.2; extra == "tabarena"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
 Requires-Dist: loguru; extra == "tabarena"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "tabarena"
-Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
+Requires-Dist: transformers; extra == "tabarena"
 Requires-Dist: huggingface-hub[torch]; extra == "tabarena"
+Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
+Requires-Dist: einx; extra == "tabarena"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
 Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: omegaconf; extra == "tabarena"
 Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.4.0b20250725; extra == "tabarena"
 Requires-Dist: pytabkit<1.6,>=1.5; extra == "tabarena"
-Requires-Dist: omegaconf; extra == "tabarena"
-Requires-Dist: transformers; extra == "tabarena"
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
-Requires-Dist: einx; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.4.0b20250726; extra == "tabarena"
+Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
 Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "tabarena"
 Provides-Extra: tabicl
 Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabicl"

{autogluon.tabular-1.4.0b20250725.dist-info → autogluon.tabular-1.4.0b20250726.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.4.0b20250725-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.4.0b20250726-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=jKQHc6rb8KlpzX1_jnH5NX-mZVITo1-rtReHD8lPZms,91
+autogluon/tabular/version.py,sha256=X677_QRDbCPTZCRCcshLbt66zrfGkhbhIjj5DPT-OPc,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
@@ -8,7 +8,7 @@ autogluon/tabular/configs/hyperparameter_configs.py,sha256=aQ1rrF8P0MX4Ic5M33O96
 autogluon/tabular/configs/presets_configs.py,sha256=KxZkUU58dxvKeY8g94gdIJkqi2eos68262-efDJVrhY,7755
 autogluon/tabular/configs/zeroshot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2023.py,sha256=6yd84vPqOk-6sLCoM_e_PlphrR2NZUjliS7L1SMKMug,29777
-autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2025.py,sha256=1EH54KvJT35xNVegh1SuvBWt0Qx07vQUdHyc10TwaOI,11966
+autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2025.py,sha256=NXwfqZLQLx4kdvRqF6deFDdhZZKxbfgpUurdB0kqOh8,11996
 autogluon/tabular/experimental/__init__.py,sha256=PpkdMSv_pPZted1XRIuzcFWKjM-66VMUukTnCcoiW0s,100
 autogluon/tabular/experimental/_scikit_mixin.py,sha256=cKeCmtURAXZnhQGrkCBw5rmACCQF7biAWTT3qX8bM2Q,2281
 autogluon/tabular/experimental/_tabular_classifier.py,sha256=7lGoFdvkHiZS3VpcXo97q4ENV9qyIVDExlWkm0wzL3s,2527
@@ -69,7 +69,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=LISemlSeyZiwC10EHfCG1V_6_lx5Z2SjLqCL6nQKf8c,12712
+autogluon/tabular/models/mitra/mitra_model.py,sha256=Ugk2-YmMRlo127jTZKNIv2qEqwNby8_nfpSspng7D3o,12253
 autogluon/tabular/models/mitra/sklearn_interface.py,sha256=Znwx1uMagauu1DwcutM_kgGY8maQrxOE0KsP1uS46qE,18751
 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
@@ -189,11 +189,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.4.0b20250725.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.4.0b20250725.dist-info/METADATA,sha256=n_wcsTvwtMvOXv7eIkWRc9qVKukcl0NhzUihuv4ZwCI,16087
-autogluon.tabular-1.4.0b20250725.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.4.0b20250725.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.4.0b20250725.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.4.0b20250725.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.4.0b20250725.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.4.0b20250725.dist-info/RECORD,,
+autogluon.tabular-1.4.0b20250726.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.4.0b20250726.dist-info/METADATA,sha256=V_CVPxxcyxysZ71Rgv0tAHgZmy9HG3srrsNYYnpd680,16087
+autogluon.tabular-1.4.0b20250726.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.4.0b20250726.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.4.0b20250726.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.4.0b20250726.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.4.0b20250726.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.4.0b20250726.dist-info/RECORD,,