gbhatia
New Contributor II

Maybe add missing: 

mlflow.set_tracking_uri("databricks")
mlflow.set_registry_uri("databricks")