diff --git a/qlib/workflow/recorder.py b/qlib/workflow/recorder.py index 9d82bf0a47..226f0476d7 100644 --- a/qlib/workflow/recorder.py +++ b/qlib/workflow/recorder.py @@ -21,6 +21,8 @@ from mlflow.store.artifact.azure_blob_artifact_repo import AzureBlobArtifactRepository logger = get_module_logger("workflow", logging.INFO) +# mlflow limits the length of log_param to 500, but this caused errors when using qrun, so we extended the mlflow limit. +mlflow.utils.validation.MAX_PARAM_VAL_LENGTH = 1000 class Recorder: diff --git a/setup.py b/setup.py index faf058d631..9ff13d4a51 100644 --- a/setup.py +++ b/setup.py @@ -62,6 +62,8 @@ def get_version(rel_path: str) -> str: "matplotlib>=3.3", "tables>=3.6.1", "pyyaml>=5.3.1", + # To ensure stable operation of the experiment manager, we have limited the version of mlflow, + # and we need to verify whether version 2.0 of mlflow can serve qlib properly. "mlflow>=1.12.1, <=1.30.0", "tqdm", "loguru",