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.htmlWhile in the latest (2.17.0) there is such https://mlflow.org/docs/2.17.0/python_api/mlflow.types.html%pip insta...
