github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-19 03:01:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change tags(str) to tags(dict) in mlflow_callback.py docs (#6473)

Browse Source 
Fixes #6472

#### Who can review?

@agola11
This commit is contained in:
ArchimedesFTW 2023-06-26 11:12:23 +02:00 committed by GitHub
parent 9187d2f3a9
commit fe941cb54a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/callbacks/mlflow_callback.py
View File

@ -118,7 +118,7 @@ class MlflowLogger:
Parameters:
name (str): Name of the run.
experiment (str): Name of the experiment.
tags (str): Tags to be attached for the run.
tags (dict): Tags to be attached for the run.
tracking_uri (str): MLflow tracking server uri.
This handler implements the helper functions to initialize,
@ -223,7 +223,7 @@ class MlflowCallbackHandler(BaseMetadataCallbackHandler, BaseCallbackHandler):
Parameters:
name (str): Name of the run.
experiment (str): Name of the experiment.
tags (str): Tags to be attached for the run.
tags (dict): Tags to be attached for the run.
tracking_uri (str): MLflow tracking server uri.
This handler will utilize the associated callback method called and formats