autogluon.common 1.2.1b20250303__py3-none-any.whl → 1.2.1b20250304__py3-none-any.whl

autogluon/common/utils/log_utils.py

@@ -193,3 +193,32 @@ def reset_logger_for_remote_call(verbosity: int):
     # Limiting the verbosity of the BaggedEnsembleModel logger to 10 to avoid
     # duplicated messages about fitting.
     set_logger_verbosity(verbosity=min(verbosity, 1), logger=bem_logger)
+
+
+def warn_if_mlflow_autologging_is_enabled(logger: Optional[logging.Logger] = None):
+    """Log a warning if MLflow autologging is enabled.
+
+    MLflow autologging monkey-patches the sklearn metrics, which leads to a PicklingError when AutoGluon models
+    that have a metric as an instance attribute are saved to disk.
+
+    Related issues:
+        - https://github.com/mlflow/mlflow/issues/6268
+        - https://github.com/autogluon/autogluon/issues/4914
+    """
+    if logger is None:
+        logger = _logger_ag
+    try:
+        import mlflow
+
+        try:
+            if not mlflow.utils.autologging_utils.autologging_is_disabled("sklearn"):
+                logger.warning(
+                    "⚠️ Warning: MLflow autologging is enabled. This will likely break AutoGluon model training. "
+                    "Please disable autologging by executing `import mlflow; mlflow.autolog(disable=True)` before importing AutoGluon."
+                )
+        except Exception:
+            # gracefully handle cases where autologging_is_disabled is not available
+            pass
+    except Exception:
+        # mlflow not installed, all good
+        pass

autogluon/common/version.py

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

{autogluon.common-1.2.1b20250303.dist-info → autogluon.common-1.2.1b20250304.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.common
-Version: 1.2.1b20250303
+Version: 1.2.1b20250304
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -42,10 +42,10 @@ Requires-Dist: boto3<2,>=1.10
 Requires-Dist: psutil<7.0.0,>=5.7.3
 Requires-Dist: tqdm<5,>=4.38
 Provides-Extra: tests
+Requires-Dist: types-requests; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-mypy; extra == "tests"
-Requires-Dist: types-setuptools; extra == "tests"
-Requires-Dist: types-requests; extra == "tests"
 
 
 

{autogluon.common-1.2.1b20250303.dist-info → autogluon.common-1.2.1b20250304.dist-info}/RECORD

@@ -1,8 +1,8 @@
-autogluon.common-1.2.1b20250303-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.common-1.2.1b20250304-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/common/__init__.py,sha256=Q0tQ4UOtUEhPpj83k_xzQ9bmtoTSvfg_3WgPi1QopJk,352
 autogluon/common/dataset.py,sha256=slfU8CxGFNJ86_2wKkT_zPMPr-iPB-sz7klz0YqJECo,1580
 autogluon/common/space.py,sha256=oRSG-uFmsLACT42mT2vRp3NSuV8MaQz6X4sYwcZruxs,5457
-autogluon/common/version.py,sha256=qmmAxNUhCru46CberfM2fZH_uvNtgol4goKRAyUQC04,91
+autogluon/common/version.py,sha256=p0UlMkkxgxdnhofvyqKPl_WnTfvNk3krUjeDbJ0df5c,91
 autogluon/common/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/common/features/feature_metadata.py,sha256=GDvZXyaNkDeuNv8omO4K7tkCPmMdpJzD5rH_hKtZD9E,25783
 autogluon/common/features/infer_types.py,sha256=MkfVDoY__0Rr__zsDSKszmvYMq5RQRmbNboTG_tm64o,6492
@@ -32,7 +32,7 @@ autogluon/common/utils/distribute_utils.py,sha256=NYEQkDxGvbQ4pJNY1E-iAJDjk4wnoG
 autogluon/common/utils/file_utils.py,sha256=S0uHTvdmljRiUIiLmwcoi399FutGpE5y1ZC1S5wv_pE,2782
 autogluon/common/utils/hyperparameter_utils.py,sha256=YYijXx7_JMmJDFwT-ZVpDJpbHjFkXka6isLzAix87Ag,2292
 autogluon/common/utils/lite.py,sha256=fRbJslvVx3vupboC1v5mnPk7fjeH2UlyIVzLskGL2-Q,317
-autogluon/common/utils/log_utils.py,sha256=jxziaOEXHBxV97qYuIRyCRTDnHsj_rwRy_19cpM3Yu8,7278
+autogluon/common/utils/log_utils.py,sha256=tx2iOBcfh52TR1wd1fHMzNPFPluESOmVUxT3_pD04s0,8435
 autogluon/common/utils/multiprocessing_utils.py,sha256=gTbOMnwOrMYaY-NJy1GVS5aIv1LEehWR8Sn57vda3kg,589
 autogluon/common/utils/nvutil.py,sha256=tVyCdUDoM7RSkO1Ek01NVRFL8bbFnQBryezYpN-BX1M,4804
 autogluon/common/utils/pandas_utils.py,sha256=sirUUQkKZsmMwmXWEUZJ1n3ZYFQWqwz4iu-tAhZdv68,2432
@@ -43,11 +43,11 @@ autogluon/common/utils/simulation_utils.py,sha256=iEZcRWzP8xLQ8racIfOGC5V_A0W_xN
 autogluon/common/utils/system_info.py,sha256=9BkFcdq1co0EMnZkk_AYdhxSEP3RqcU6uIp-iYx9bdc,4479
 autogluon/common/utils/try_import.py,sha256=k_d9RvHs_6ShL5gN7OuVtNHhZZ9BOxz4t6piOkfpug4,7499
 autogluon/common/utils/utils.py,sha256=H0gjBbiFBPuhnlHpirN3nU7WehoTZCv-231tb1oIvMI,7971
-autogluon.common-1.2.1b20250303.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.common-1.2.1b20250303.dist-info/METADATA,sha256=jmZtVrOWnPxJNKbg1mPomf6Hibfxzgn_at8JZQXkr1s,11658
-autogluon.common-1.2.1b20250303.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.common-1.2.1b20250303.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.common-1.2.1b20250303.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250303.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250303.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.common-1.2.1b20250303.dist-info/RECORD,,
+autogluon.common-1.2.1b20250304.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.common-1.2.1b20250304.dist-info/METADATA,sha256=Rmn2XdXUvJS0IIfwolxBzU_5Y_iEMk6QebHdQ5Pfahw,11658
+autogluon.common-1.2.1b20250304.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.common-1.2.1b20250304.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.common-1.2.1b20250304.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250304.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250304.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.common-1.2.1b20250304.dist-info/RECORD,,