autogluon.tabular 1.4.1b20250818__tar.gz → 1.4.1b20250819__tar.gz

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/PKG-INFO

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

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/src/autogluon/tabular/models/mitra/mitra_model.py

@@ -218,6 +218,31 @@ class MitraModel(AbstractModel):
             model._weights_saved = False
         return model
 
+    @classmethod
+    def download_weights(cls, repo_id: str):
+        """
+        Download weights for Mitra from HuggingFace from `repo_id`.
+        Requires an internet connection.
+        """
+        from huggingface_hub import hf_hub_download
+        hf_hub_download(repo_id=repo_id, filename="config.json")
+        hf_hub_download(repo_id=repo_id, filename="model.safetensors")
+
+    @classmethod
+    def download_default_weights(cls):
+        """
+        Download default weights for Mitra from HuggingFace.
+        Includes both classifier and regressor weights.
+
+        This is useful to call when building a docker image to avoid having to download Mitra weights for each instance.
+        This is also useful for benchmarking as a first sanity check
+        to avoid HuggingFace potentially blocking the download.
+
+        Requires an internet connection.
+        """
+        cls.download_weights(repo_id="autogluon/mitra-classifier")
+        cls.download_weights(repo_id="autogluon/mitra-regressor")
+
     @classmethod
     def supported_problem_types(cls) -> Optional[List[str]]:
         return ["binary", "multiclass", "regression"]

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/src/autogluon/tabular/predictor/predictor.py

@@ -165,6 +165,10 @@ class TabularPredictor:
             trainer_type : AbstractTabularTrainer, default = AutoTrainer
                 A class inheriting from `AbstractTabularTrainer` that controls training/ensembling of many models.
                 If you don't know what this is, keep it as the default.
+        default_base_path : str | Path | None, default = None
+            A default base path to use for the time-stamped folder if `path` is None.
+            If None, defaults to `AutogluonModels`. Only used if `path` is None, and thus
+            only used for local paths, not s3 paths.
     """
 
     Dataset = TabularDataset
@@ -201,7 +205,7 @@ class TabularPredictor:
                 f"We do not recommend specifying weight_evaluation when sample_weight='{self.sample_weight}', instead specify appropriate eval_metric."
             )
         self._validate_init_kwargs(kwargs)
-        path = setup_outputdir(path)
+        path = setup_outputdir(path=path, default_base_path=kwargs.get("default_base_path"))
 
         learner_type = kwargs.get("learner_type", DefaultLearner)
         learner_kwargs = kwargs.get("learner_kwargs", dict())
@@ -5061,6 +5065,7 @@ class TabularPredictor:
             "learner_type",
             "learner_kwargs",
             "quantile_levels",
+            "default_base_path",
         }
         invalid_keys = []
         for key in kwargs:

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/src/autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.4.1b20250818"
+__version__ = "1.4.1b20250819"
 __lite__ = False

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.4.1b20250818 → autogluon.tabular-1.4.1b20250819}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,24 +3,24 @@ 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.1b20250818
-autogluon.features==1.4.1b20250818
+autogluon.core==1.4.1b20250819
+autogluon.features==1.4.1b20250819
 
 [all]
-torch<2.8,>=2.6
-lightgbm<4.7,>=4.0
-numpy<2.3.0,>=1.25
-einops<0.9,>=0.7
-xgboost<3.1,>=2.0
+omegaconf
+huggingface_hub[torch]
 einx
-transformers
+numpy<2.3.0,>=1.25
 catboost<1.3,>=1.2
-huggingface_hub[torch]
+einops<0.9,>=0.7
+autogluon.core[all]==1.4.1b20250819
+torch<2.8,>=2.6
 loguru
+lightgbm<4.7,>=4.0
 spacy<3.9
-autogluon.core[all]==1.4.1b20250818
+xgboost<3.1,>=2.0
 fastai<2.9,>=2.3.1
-omegaconf
+transformers
 
 [all:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0
@@ -53,7 +53,7 @@ huggingface_hub[torch]
 einops<0.9,>=0.7
 
 [ray]
-autogluon.core[all]==1.4.1b20250818
+autogluon.core[all]==1.4.1b20250819
 
 [realmlp]
 pytabkit<1.7,>=1.6
@@ -73,23 +73,23 @@ onnx<1.18.0,>=1.13.0
 onnx<1.16.2,>=1.13.0
 
 [tabarena]
-tabicl<0.2,>=0.1.3
-tabpfn<2.2,>=2.0.9
-torch<2.8,>=2.6
-fastai<2.9,>=2.3.1
-lightgbm<4.7,>=4.0
+omegaconf
+huggingface_hub[torch]
+einx
 numpy<2.3.0,>=1.25
+catboost<1.3,>=1.2
 einops<0.9,>=0.7
+autogluon.core[all]==1.4.1b20250819
+tabicl<0.2,>=0.1.3
+torch<2.8,>=2.6
+tabpfn<2.2,>=2.0.9
 pytabkit<1.7,>=1.6
-xgboost<3.1,>=2.0
-einx
-transformers
-catboost<1.3,>=1.2
 loguru
+lightgbm<4.7,>=4.0
 spacy<3.9
-autogluon.core[all]==1.4.1b20250818
-huggingface_hub[torch]
-omegaconf
+xgboost<3.1,>=2.0
+fastai<2.9,>=2.3.1
+transformers
 
 [tabarena:platform_system == "Windows" and python_version == "3.9"]
 blis<1.2.1,>=0.7.0