autogluon.common 1.2.1b20250302__tar.gz → 1.2.1b20250304__tar.gz

{autogluon.common-1.2.1b20250302 → autogluon.common-1.2.1b20250304}/PKG-INFO

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

{autogluon.common-1.2.1b20250302 → autogluon.common-1.2.1b20250304}/src/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-1.2.1b20250302 → autogluon.common-1.2.1b20250304}/src/autogluon/common/version.py

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

{autogluon.common-1.2.1b20250302 → autogluon.common-1.2.1b20250304}/src/autogluon.common.egg-info/PKG-INFO

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

{autogluon.common-1.2.1b20250302 → autogluon.common-1.2.1b20250304}/src/autogluon.common.egg-info/requires.txt

@@ -5,7 +5,7 @@ psutil<7.0.0,>=5.7.3
 tqdm<5,>=4.38
 
 [tests]
+types-requests
 types-setuptools
 pytest
-types-requests
 pytest-mypy