From 5dc15ce6e280c3a22c839757b7b5d4a4761b5c45 Mon Sep 17 00:00:00 2001 From: kouzhenzhong Date: Tue, 16 Jun 2020 21:30:43 +0800 Subject: [PATCH] Dataset: convert schema to json in serializer --- mindspore/dataset/engine/serializer_deserializer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/dataset/engine/serializer_deserializer.py b/mindspore/dataset/engine/serializer_deserializer.py index 812ec149f0..0e39086b02 100644 --- a/mindspore/dataset/engine/serializer_deserializer.py +++ b/mindspore/dataset/engine/serializer_deserializer.py @@ -156,6 +156,9 @@ def traverse(node): serialize_operations(node_repr, k, v) elif k == 'sampler': serialize_sampler(node_repr, v) + # return schema json str if its type is mindspore.dataset.Schema + elif k == 'schema' and isinstance(v, de.Schema): + node_repr[k] = v.to_json() elif k in set(['schema', 'dataset_files', 'dataset_dir', 'schema_file_path']): expand_path(node_repr, k, v) else: