viksuper555
New Contributor II

Upgrade the version of the mlflow package. In 2.7.1 there is no such parameter. 
https://mlflow.org/docs/2.7.1/python_api/mlflow.types.html
While in the latest (2.17.0) there is such https://mlflow.org/docs/2.17.0/python_api/mlflow.types.html

%pip install mlflow==2.17.0
dbutils.library.restartPython()