"ML Quickstart: Model Training" databrick notebook, module not found issue

DamienSicard
New Contributor III

Hi everybody,

Before jump in my own ML experiment on databricks ML platform, I take a look inside some databricks notebook.

With the first ML notebook, named "ML Quickstart: Model training" it doesn't find the library mlflow (cf image below).

I assume this is a "trivial issue", that could be solved by a command like:

!pip install mlflow

But, I would like to notice the community about this issue.

Does somebody encounter the following issue in this notebook ?

Thanks for your time.

Best,

Damien

mlflow not found

shan_chandra
Databricks Employee
Databricks Employee

mlflow is an inbuilt library with DBR ML run time such as DBR 9.1 LTS ML. Kindly use cluster with ML runtime.

View solution in original post