Import mlflow integration from new air.integrations module (#264)

justinvyu · web-flow · commit 95071b74a1c1 · 2023-06-30T14:27:25.000-07:00
* Import from new air.integrations module Signed-off-by: Justin Yu <justinvyu@anyscale.com> * dummy commit to relaunch ci Signed-off-by: Justin Yu <justinvyu@anyscale.com> * dummy commit to launch CI Signed-off-by: Justin Yu <justinvyu@anyscale.com> * Use master wheel to make sure CI passes Signed-off-by: Justin Yu <justinvyu@anyscale.com> * Revert "Use master wheel to make sure CI passes" This reverts commit e408ed8. Signed-off-by: Justin Yu <justinvyu@anyscale.com> --------- Signed-off-by: Justin Yu <justinvyu@anyscale.com>
diff --git a/tune_sklearn/utils.py b/tune_sklearn/utils.py
@@ -14,7 +14,7 @@
 from ray.tune.logger import (Logger, JsonLoggerCallback, CSVLoggerCallback,
                              TBXLoggerCallback, LegacyLoggerCallback,
                              LoggerCallback)
-from ray.tune.integration.mlflow import MLflowLoggerCallback
+from ray.air.integrations.mlflow import MLflowLoggerCallback
 
 try:
     from ray.tune.stopper import MaximumIterationStopper
