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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -275,7 +275,7 @@ class MlflowCallbackHandler(BaseMetadataCallbackHandler, BaseCallbackHandler):
tags: Optional[Dict] = None, tags: Optional[Dict] = None,
tracking_uri: Optional[str] = None, tracking_uri: Optional[str] = None,
run_id: Optional[str] = None, run_id: Optional[str] = None,
artifacts_dir: Optional[str] = None, artifacts_dir: str = "",
) -> None: ) -> None:
"""Initialize callback handler.""" """Initialize callback handler."""
import_pandas() import_pandas()