Skip to content
This repository has been archived by the owner on Nov 14, 2023. It is now read-only.

Import mlflow integration from new air.integrations module #264

Merged
merged 5 commits into from
Jun 30, 2023

Conversation

justinvyu
Copy link
Contributor

Ray has some unit-tests depending on this package. The ray.tune.integration package is getting hard-deprecated in Ray 2.6, so this needs package needs to import from the new location at ray.air.integrations to unblock that.

In particular, it's this user guide doctest that would fail: https://docs.ray.io/en/latest/tune/examples/tune-sklearn.html

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
@Yard1
Copy link
Member

Yard1 commented May 25, 2023

Actually, can we have a try...except import here? @justinvyu

Copy link
Member

@Yard1 Yard1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

see above

@justinvyu
Copy link
Contributor Author

I think the MLFlowLoggerCallback already try catches if mlflow is not installed.

@matthewdeng
Copy link

@Yard1 by try/except do you mean something like:

try import from air --> except import from tune

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
@justinvyu
Copy link
Contributor Author

No need to try catch since tune_sklearn depends on ray>=2.0, and the module was already moved to air.integrations at 2.0. Relaunching the CI, then we can merge after it passes.

@justinvyu
Copy link
Contributor Author

justinvyu commented Jun 28, 2023

CI for latest Ray release is failing due to a fix that did not make the branch cut but will be fixed on 2.6.

CI for master is failing due to a similar issue as this flaky example on ray: ray-project/ray#35428

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
Signed-off-by: Justin Yu <justinvyu@anyscale.com>
@justinvyu
Copy link
Contributor Author

This reverts commit e408ed8.

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
@Yard1 Yard1 merged commit 95071b7 into ray-project:master Jun 30, 2023
@justinvyu justinvyu deleted the update_mlflow_import branch July 10, 2023 18:39
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants