提交 ef698a93 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2185 Dataset: convert schema to json in serializer

Merge pull request !2185 from kouzhenzhong/wide_deep_dataset_graph
...@@ -156,6 +156,9 @@ def traverse(node): ...@@ -156,6 +156,9 @@ def traverse(node):
serialize_operations(node_repr, k, v) serialize_operations(node_repr, k, v)
elif k == 'sampler': elif k == 'sampler':
serialize_sampler(node_repr, v) 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']): elif k in set(['schema', 'dataset_files', 'dataset_dir', 'schema_file_path']):
expand_path(node_repr, k, v) expand_path(node_repr, k, v)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册