autogluon.tabular 1.4.1b20250818__py3-none-any.whl → 1.4.1b20250820__py3-none-any.whl

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/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/version.py

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

{autogluon.tabular-1.4.1b20250818.dist-info → autogluon.tabular-1.4.1b20250820.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.4.1b20250818
+Version: 1.4.1b20250820
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,23 +41,23 @@ 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.1b20250818
-Requires-Dist: autogluon.features==1.4.1b20250818
+Requires-Dist: autogluon.core==1.4.1b20250820
+Requires-Dist: autogluon.features==1.4.1b20250820
 Provides-Extra: all
-Requires-Dist: torch<2.8,>=2.6; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
-Requires-Dist: einx; extra == "all"
-Requires-Dist: transformers; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: spacy<3.9; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: einx; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: transformers; extra == "all"
+Requires-Dist: omegaconf; extra == "all"
 Requires-Dist: loguru; extra == "all"
-Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: autogluon.core[all]==1.4.1b20250818; extra == "all"
+Requires-Dist: autogluon.core[all]==1.4.1b20250820; extra == "all"
 Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: omegaconf; extra == "all"
+Requires-Dist: torch<2.8,>=2.6; 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"
@@ -80,7 +80,7 @@ Requires-Dist: transformers; extra == "mitra"
 Requires-Dist: huggingface-hub[torch]; extra == "mitra"
 Requires-Dist: einops<0.9,>=0.7; extra == "mitra"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.4.1b20250818; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.4.1b20250820; extra == "ray"
 Provides-Extra: realmlp
 Requires-Dist: pytabkit<1.7,>=1.6; extra == "realmlp"
 Provides-Extra: skex
@@ -92,23 +92,23 @@ 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: tabicl<0.2,>=0.1.3; extra == "tabarena"
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
-Requires-Dist: torch<2.8,>=2.6; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "tabarena"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "tabarena"
-Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
 Requires-Dist: pytabkit<1.7,>=1.6; extra == "tabarena"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "tabarena"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
 Requires-Dist: einx; extra == "tabarena"
+Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabarena"
+Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
 Requires-Dist: transformers; extra == "tabarena"
-Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
+Requires-Dist: torch<2.8,>=2.6; extra == "tabarena"
+Requires-Dist: omegaconf; extra == "tabarena"
 Requires-Dist: loguru; extra == "tabarena"
-Requires-Dist: spacy<3.9; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.4.1b20250818; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.4.1b20250820; extra == "tabarena"
 Requires-Dist: huggingface-hub[torch]; extra == "tabarena"
-Requires-Dist: omegaconf; extra == "tabarena"
+Requires-Dist: einops<0.9,>=0.7; 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.1b20250818.dist-info → autogluon.tabular-1.4.1b20250820.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.4.1b20250818-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.4.1b20250820-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=gfOGF027svt-J46kXd4_hPGSL8Tf07_cRlCOodxTxVM,91
+autogluon/tabular/version.py,sha256=S2ZZQJcoKT498Nmq-Gu28zMiZsIYa6iD670boLDdwfg,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=Rby5gRhuY5IlZWdKbtsmzbSt948B97qxwQ2f1MbH_38,21070
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -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=9c0qwFnKl5WgdSuMMjluGKCSh0BePWT9_RWPtVb7Rcc,12435
+autogluon/tabular/models/mitra/mitra_model.py,sha256=lHcOtaQwIy4EpgK4TAiuRqQQFhBN3NzSdejFjkNpZPg,13447
 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
@@ -173,7 +173,7 @@ autogluon/tabular/models/xt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 autogluon/tabular/models/xt/xt_model.py,sha256=qOHJ5h1lHI7uYJfbl0BWm-29R3MNp2WeZB9ptcq5Xis,1003
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=akgjBQMpyuuvxSGDhaTccJmYhqA8dKYFKNmn88c8VIg,360587
+autogluon/tabular/predictor/predictor.py,sha256=CRP98NWyRnWYHjvWpqSpfWo3CX5j170M6GXXe2lDw-g,360967
 autogluon/tabular/registry/__init__.py,sha256=vZpzX4Xve7bfA9crt5LxjgQv9PPfxbi1E1U6Im0Y_xU,93
 autogluon/tabular/registry/_ag_model_registry.py,sha256=Aa-o_KZZiroPBpvZozIBXOlWYvQJN-MVsl_Gl66gkE8,1550
 autogluon/tabular/registry/_model_registry.py,sha256=Rl8Q7BLzaif4hxNxJF20xGE02vrWwh2ZuUaTmA-UJnE,6824
@@ -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.1b20250818.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.4.1b20250818.dist-info/METADATA,sha256=esQVnuXwDNKk-UqUNeTGwM8RL9gjRCokDnOi5bgUYz0,16238
-autogluon.tabular-1.4.1b20250818.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.4.1b20250818.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.4.1b20250818.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.4.1b20250818.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.4.1b20250818.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.4.1b20250818.dist-info/RECORD,,
+autogluon.tabular-1.4.1b20250820.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.4.1b20250820.dist-info/METADATA,sha256=oxY-iF3arfj3QFp_5xjlbckx6lVKwN-yDcncp2EJUxo,16238
+autogluon.tabular-1.4.1b20250820.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.4.1b20250820.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.4.1b20250820.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.4.1b20250820.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.4.1b20250820.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.4.1b20250820.dist-info/RECORD,,