From fe941cb54a80976bfc7575ce59a518ae428801ee Mon Sep 17 00:00:00 2001
From: ArchimedesFTW <55234938+ArchimedesFTW@users.noreply.github.com>
Date: Mon, 26 Jun 2023 11:12:23 +0200
Subject: [PATCH] Change tags(str) to tags(dict) in mlflow_callback.py docs
 (#6473)

Fixes #6472

#### Who can review?

@agola11
---
 langchain/callbacks/mlflow_callback.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/langchain/callbacks/mlflow_callback.py b/langchain/callbacks/mlflow_callback.py
index 34b05e0e3..8bae7739f 100644
--- a/langchain/callbacks/mlflow_callback.py
+++ b/langchain/callbacks/mlflow_callback.py
@@ -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