Hi @Maria Bruevich​ ,

From the error description, it looks like the mlflow library is not present. You can use ML cluster as these type of cluster already have mlflow library. Please check the below document:

• https://docs.databricks.com/release-notes/runtime/7.3ml.html

Or else, we will need to install the required library into the existing cluster.

Below document will help to install the library:

• https://docs.databricks.com/libraries/cluster-libraries.html

Please let us know if this helps.