From 5c6e12375732ebdaa971401f936b2f7783d61679 Mon Sep 17 00:00:00 2001 From: Serena Ruan <82044803+serena-ruan@users.noreply.github.com> Date: Tue, 23 Jan 2024 19:09:02 -0800 Subject: [PATCH] community[patch]: Fix MlflowCallback with none artifacts_dir (#16487) --- libs/community/langchain_community/callbacks/mlflow_callback.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/community/langchain_community/callbacks/mlflow_callback.py b/libs/community/langchain_community/callbacks/mlflow_callback.py index 577532a3616..4070d17d61c 100644 --- a/libs/community/langchain_community/callbacks/mlflow_callback.py +++ b/libs/community/langchain_community/callbacks/mlflow_callback.py @@ -275,7 +275,7 @@ class MlflowCallbackHandler(BaseMetadataCallbackHandler, BaseCallbackHandler): tags: Optional[Dict] = None, tracking_uri: Optional[str] = None, run_id: Optional[str] = None, - artifacts_dir: Optional[str] = None, + artifacts_dir: str = "", ) -> None: """Initialize callback handler.""" import_pandas()