community[patch]: Fix MlflowCallback with none artifacts_dir (#16487)

This commit is contained in:
Serena Ruan
2024-01-23 19:09:02 -08:00
committed by GitHub
parent 0e2e7d8b83
commit 5c6e123757

View File

@@ -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()