No Module named 'mlflow'

duck_butter123
New Contributor II

I new to the scalable machine learning with apache spark course. I am in the notebook ML 00a - Install Datasets it includes one cell (attached) which throws an error 'no module named 'mlflow''. It attempts to run the Classroom-Setup file. Error is thrown on cmd 5 (attached). What can I do to remedy this? Classroom-Setup code + errorML 00a - Install Dataset Error

Hubert-Dudek
Databricks MVP

Please use Machine Learning Runtime for your cluster https://docs.databricks.com/runtime/mlruntime.html


My blog: https://databrickster.medium.com/

View solution in original post

Anonymous
Not applicable

Hubert is correct, make sure to use the ML Runtime. I attached a picture from the cluster creation page.

Prabakar
Databricks Employee
Databricks Employee

We have ML DBR version to be used for ML related workloads. So it is adviced to use ml dbr instead of standard dbr.​

Anonymous
Not applicable

@Myles Pember​ I hope the suggestions above helped out! If so, please select one as 'best' for us!

If you still need assistance, let us know! 😊