autogluon.tabular 1.2.1b20250303__py3-none-any.whl → 1.2.1b20250305__py3-none-any.whl

autogluon/tabular/models/automm/ft_transformer.py

@@ -73,14 +73,14 @@ class FTTransformerModel(MultiModalPredictorModel):
             "env.batch_size": 128,
             "env.per_gpu_batch_size": 128,
             "env.num_workers": 0,
-            "env.num_workers_evaluation": 0,
-            "optimization.max_epochs": 2000,  # Specify a large value to train until convergence
-            "optimization.weight_decay": 1.0e-5,
-            "optimization.lr_choice": None,
-            "optimization.lr_schedule": "polynomial_decay",
-            "optimization.warmup_steps": 0.0,
-            "optimization.patience": 20,
-            "optimization.top_k": 3,
+            "env.num_workers_inference": 0,
+            "optim.max_epochs": 2000,  # Specify a large value to train until convergence
+            "optim.weight_decay": 1.0e-5,
+            "optim.lr_choice": None,
+            "optim.lr_schedule": "polynomial_decay",
+            "optim.warmup_steps": 0.0,
+            "optim.patience": 20,
+            "optim.top_k": 3,
             "_max_features": 300,  # FIXME: This is a hack, move to AG_ARGS_FIT for v0.7
         }
         for param, val in default_params.items():

autogluon/tabular/predictor/predictor.py

@@ -21,7 +21,7 @@ from autogluon.common.loaders import load_json
 from autogluon.common.savers import save_json
 from autogluon.common.utils.file_utils import get_directory_size, get_directory_size_per_file
 from autogluon.common.utils.hyperparameter_utils import get_hyperparameter_str_deprecation_msg, is_advanced_hyperparameter_format
-from autogluon.common.utils.log_utils import add_log_to_file, set_logger_verbosity
+from autogluon.common.utils.log_utils import add_log_to_file, set_logger_verbosity, warn_if_mlflow_autologging_is_enabled
 from autogluon.common.utils.pandas_utils import get_approximate_df_mem_usage
 from autogluon.common.utils.system_info import get_ag_system_info
 from autogluon.common.utils.try_import import try_import_ray
@@ -1051,6 +1051,7 @@ class TabularPredictor:
 
         verbosity = kwargs.get("verbosity", self.verbosity)
         set_logger_verbosity(verbosity)
+        warn_if_mlflow_autologging_is_enabled(logger=logger)
 
         if verbosity >= 2:
             if verbosity == 2:

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250303"
+__version__ = "1.2.1b20250305"
 __lite__ = False

{autogluon.tabular-1.2.1b20250303.dist-info → autogluon.tabular-1.2.1b20250305.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250303
+Version: 1.2.1b20250305
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,19 +41,19 @@ Requires-Dist: scipy<1.16,>=1.5.4
 Requires-Dist: pandas<2.3.0,>=2.0.0
 Requires-Dist: scikit-learn<1.5.3,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.2.1b20250303
-Requires-Dist: autogluon.features==1.2.1b20250303
+Requires-Dist: autogluon.core==1.2.1b20250305
+Requires-Dist: autogluon.features==1.2.1b20250305
 Provides-Extra: all
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: torch<2.6,>=2.2; extra == "all"
-Requires-Dist: huggingface-hub[torch]; extra == "all"
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
-Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
+Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
+Requires-Dist: spacy<3.8; extra == "all"
+Requires-Dist: autogluon.core[all]==1.2.1b20250305; extra == "all"
 Requires-Dist: fastai<2.8,>=2.3.1; extra == "all"
+Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250303; extra == "all"
-Requires-Dist: spacy<3.8; extra == "all"
-Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
+Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: torch<2.6,>=2.2; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -66,7 +66,7 @@ Requires-Dist: imodels<1.4.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.6,>=4.0; extra == "lightgbm"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.2.1b20250303; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250305; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.1,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

{autogluon.tabular-1.2.1b20250303.dist-info → autogluon.tabular-1.2.1b20250305.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250303-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250305-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=qmmAxNUhCru46CberfM2fZH_uvNtgol4goKRAyUQC04,91
+autogluon/tabular/version.py,sha256=hjp__CpU1tLt2wGIEfd4tVSKrES_x0vHTkZdLuoYGnE,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=Pb2aW9Z9w77pYKPRVZ3nBzHY3KJaiEJSJ747zZcJIVk,21132
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -22,7 +22,7 @@ autogluon/tabular/models/_utils/rapids_utils.py,sha256=gbej9Hjn4alCWZuGN9sOLXMMA
 autogluon/tabular/models/_utils/torch_utils.py,sha256=dxs_KMMAOmNkRNjYf_hrzqaHIfkqn1xoKRKqCFbQ1Rk,537
 autogluon/tabular/models/automm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/automm/automm_model.py,sha256=_FPkXiHjowuc_KTKVkrzstaj1pecPDxIAZfw0LVjbJ0,11220
-autogluon/tabular/models/automm/ft_transformer.py,sha256=8T80u3SbWr2-0KagOXhYpZkTvFIbRF91rE8sVZN0w_0,3901
+autogluon/tabular/models/automm/ft_transformer.py,sha256=ryaMWA4tPmx_8_qdqOyzbIW6f5cBbLSwdtiHf9kfs80,3851
 autogluon/tabular/models/catboost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/catboost/callbacks.py,sha256=l8x17n_w7oEFs-iDECSdBKZ89yW5g1z-zvj4XLgQPkw,7098
 autogluon/tabular/models/catboost/catboost_model.py,sha256=UmM7vnDAeQnIXRZ2MXYMP0mwB3Vw_oNhUGmRrxBJ5sw,17380
@@ -144,7 +144,7 @@ autogluon/tabular/models/xt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 autogluon/tabular/models/xt/xt_model.py,sha256=vXy1HQ06fG1YjPTf0-THBKgcA9Edw1XhmBC4aOB3qVA,967
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=A86C_9ixByrSrcU-Bxy9eTTZ7nCnIvRDN2uY-0ANLVY,357082
+autogluon/tabular/predictor/predictor.py,sha256=jOkpypHAPrL2nsI4iypVkZV90TpMORK-G_Ixr3Kw3XQ,357182
 autogluon/tabular/register/__init__.py,sha256=7CLOTWIUho0wi4eAwhYJ5Y0PfvNCWKnRwlw3bwYoTNE,93
 autogluon/tabular/register/_ag_model_register.py,sha256=tMr-QgxgCE49tdThdSFOZaJg2D9ckDh6fiR5K4cRtvk,1564
 autogluon/tabular/register/_model_register.py,sha256=jqSg0d89dXAAcp-OT4II90ce994ByKMMzAYmpkyaRbI,6824
@@ -156,11 +156,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=bTPGPyz07a7GG6327yO6ry
 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.2.1b20250303.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250303.dist-info/METADATA,sha256=A-4AXu8KMq_AxKcZwcIzWLbGLy8f-qKJ4HFJ7lZOZEg,14386
-autogluon.tabular-1.2.1b20250303.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250303.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250303.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250303.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250303.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250303.dist-info/RECORD,,
+autogluon.tabular-1.2.1b20250305.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.2.1b20250305.dist-info/METADATA,sha256=cakvMqjhws1tFxCEq1ooyd8wqFoimmrVHE27YYzoUM4,14386
+autogluon.tabular-1.2.1b20250305.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.2.1b20250305.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.2.1b20250305.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250305.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250305.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.2.1b20250305.dist-info/RECORD,,