WiliamRosa
Databricks Partner

Why MLflow and Databricks library versions?

Did you try increasing the MLflow artifact upload/download timeout, for example:

import os
os.environ["MLFLOW_ARTIFACT_UPLOAD_DOWNLOAD_TIMEOUT"] = "120"

Wiliam Rosa
Data Engineer | Machine Learning Engineer
LinkedIn: linkedin.com/in/wiliamrosa