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

[DamienSicard]

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

[shan_chandra · Accepted solution]

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