From 7f1a5448dc78cb02ce482f62da2631e72ce98f6b Mon Sep 17 00:00:00 2001
From: luopengting <luopengting@huawei.com>
Date: Thu, 28 Jan 2021 16:42:38 +0800
Subject: [PATCH] support recording dict as string for dataset graph

---
 mindspore/train/callback/_dataset_graph.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mindspore/train/callback/_dataset_graph.py b/mindspore/train/callback/_dataset_graph.py
index 3630c389e4..761e47dd9b 100644
--- a/mindspore/train/callback/_dataset_graph.py
+++ b/mindspore/train/callback/_dataset_graph.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 # ============================================================================
 """Define dataset graph related operations."""
+import json
 from importlib import import_module
 
 from mindspore import log as logger
@@ -124,6 +125,11 @@ class DatasetGraph:
             if value:
                 replace_value_list = list(map(lambda x: "" if x is None else x, value))
                 message.mapStrList[key].strValue.extend(replace_value_list)
+        elif isinstance(value, dict):
+            try:
+                message.mapStr[key] = json.dumps(value)
+            except TypeError as exo:
+                logger.warning("Transform the value of parameter %r to string failed. Detail: %s.", key, str(exo))
         elif value is None:
             message.mapStr[key] = "None"
         else: