From 3a5a2d2f34c14b4dd9289ed7062b4e909de546f7 Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Wed, 28 Feb 2024 23:28:29 +0530 Subject: [PATCH] Fix `use_mlflow` to be bool instead of str (#1344) --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 0a39f269a..262e8928a 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -322,7 +322,7 @@ class ModelOutputConfig(BaseModel): class MLFlowConfig(BaseModel): """mlflow configuration subset""" - use_mlflow: Optional[str] = None + use_mlflow: Optional[bool] = None mlflow_tracking_uri: Optional[str] = None mlflow_experiment_name: Optional[str] = None hf_mlflow_log_artifacts: Optional[bool] = None