A bug occurs when converting from mlflow to aim. - Githubissues

aimhubio / aim

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

https://aimstack.io

Apache License 2.0

5.16k stars 316 forks source link

A bug occurs when converting from mlflow to aim. #2984

Open Mai0313 opened 1 year ago

Mai0313 commented 1 year ago

🐛 Bug

Due to mlflow update PR#6171, list_experiments has been replaced to search_experiments; it would be great if you guys replace it.

src/aimcore/cli/convert/processors/mlflow.py

To reproduce

aim convert mlflow --tracking_uri 'path/to/folder/mlruns'

Expected behavior

Original error code:

AttributeError: 'MlflowClient' object has no attribute 'list_experiments'

Once you replace list_experiments to search_experiments, it will run perfectly.

Environment

Aim Version (e.g., 3.0.1): 3.17.5
MLflow: 2.5.0
Python version: 3.9.15
pip version: 23.2.1
OS (e.g., Linux): Linux Ubuntu 18.04.6 LTS x86_64

Additional context

I can create a PR if you guys want me to do so, or whatever you guys like.