Skip to content

Commit fe941cb

Browse files
Change tags(str) to tags(dict) in mlflow_callback.py docs (#6473)
Fixes #6472 #### Who can review? @agola11
1 parent 9187d2f commit fe941cb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

langchain/callbacks/mlflow_callback.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ class MlflowLogger:
118118
Parameters:
119119
name (str): Name of the run.
120120
experiment (str): Name of the experiment.
121-
tags (str): Tags to be attached for the run.
121+
tags (dict): Tags to be attached for the run.
122122
tracking_uri (str): MLflow tracking server uri.
123123
124124
This handler implements the helper functions to initialize,
@@ -223,7 +223,7 @@ class MlflowCallbackHandler(BaseMetadataCallbackHandler, BaseCallbackHandler):
223223
Parameters:
224224
name (str): Name of the run.
225225
experiment (str): Name of the experiment.
226-
tags (str): Tags to be attached for the run.
226+
tags (dict): Tags to be attached for the run.
227227
tracking_uri (str): MLflow tracking server uri.
228228
229229
This handler will utilize the associated callback method called and formats

0 commit comments

Comments
 (0)