From 98c1f6f876c6d34cada8dec47f08d9f4c5df4148 Mon Sep 17 00:00:00 2001 From: kouzhenzhong Date: Wed, 17 Jun 2020 20:13:24 +0800 Subject: [PATCH] Dataset: serializer return padd_sample as json str for record --- 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 0e39086b02..d0399d6bea 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) + elif k == 'padded_sample' and v: + v1 = {key: value for key, value in v.items() if not isinstance(value, bytes)} + node_repr[k] = json.dumps(v1, indent=2) # 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()