autogluon.tabular 1.4.0b20250725__tar.gz → 1.4.0b20250726__tar.gz

{autogluon.tabular-1.4.0b20250725 → autogluon.tabular-1.4.0b20250726}/PKG-INFO

@@ -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

{autogluon.tabular-1.4.0b20250725 → autogluon.tabular-1.4.0b20250726}/src/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-1.4.0b20250725 → autogluon.tabular-1.4.0b20250726}/src/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-1.4.0b20250725 → autogluon.tabular-1.4.0b20250726}/src/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 → autogluon.tabular-1.4.0b20250726}/src/autogluon.tabular.egg-info/PKG-INFO

@@ -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

{autogluon.tabular-1.4.0b20250725 → autogluon.tabular-1.4.0b20250726}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,23 +3,23 @@ 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.0b20250725
-autogluon.features==1.4.0b20250725
+autogluon.core==1.4.0b20250726
+autogluon.features==1.4.0b20250726
 
 [all]
+lightgbm<4.7,>=4.0
 xgboost<3.1,>=2.0
 torch<2.8,>=2.2
-einx
-lightgbm<4.7,>=4.0
-fastai<2.9,>=2.3.1
 loguru
 numpy<2.3.0,>=1.25
-catboost<1.3,>=1.2
+transformers
 huggingface_hub[torch]
+einx
+fastai<2.9,>=2.3.1
 spacy<3.9
-autogluon.core[all]==1.4.0b20250725
 omegaconf
-transformers
+autogluon.core[all]==1.4.0b20250726
+catboost<1.3,>=1.2
 
 [all:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0
@@ -51,7 +51,7 @@ transformers
 huggingface_hub[torch]
 
 [ray]
-autogluon.core[all]==1.4.0b20250725
+autogluon.core[all]==1.4.0b20250726
 
 [realmlp]
 pytabkit<1.6,>=1.5
@@ -71,22 +71,22 @@ onnx<1.18.0,>=1.13.0
 onnx<1.16.2,>=1.13.0
 
 [tabarena]
+lightgbm<4.7,>=4.0
 xgboost<3.1,>=2.0
 torch<2.8,>=2.2
-lightgbm<4.7,>=4.0
-fastai<2.9,>=2.3.1
 loguru
 numpy<2.3.0,>=1.25
-catboost<1.3,>=1.2
+transformers
 huggingface_hub[torch]
+tabpfn<2.2,>=2.0.9
+einx
+fastai<2.9,>=2.3.1
 spacy<3.9
+omegaconf
 tabicl<0.2,>=0.1.3
-autogluon.core[all]==1.4.0b20250725
 pytabkit<1.6,>=1.5
-omegaconf
-transformers
-tabpfn<2.2,>=2.0.9
-einx
+autogluon.core[all]==1.4.0b20250726
+catboost<1.3,>=1.2
 
 [tabarena:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0