From fa2a8540575155b8777b39e29e8bcee855c8ff71 Mon Sep 17 00:00:00 2001
From: ms_yan <6576637+ms_yan@user.noreply.gitee.com>
Date: Sat, 30 Jan 2021 11:32:40 +0800
Subject: [PATCH] fix GeneratorDataset unable to do serialize

---
 mindspore/dataset/engine/datasets.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py
index f04c9a42a1..b95dbb77b1 100644
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -214,7 +214,8 @@ class Dataset:
         Returns:
             str, JSON string of the pipeline.
         """
-        return json.loads(self.parse_tree().to_json(filename))
+        ir_tree, _ = self.create_ir_tree()
+        return json.loads(ir_tree.to_json(filename))
 
     @check_bucket_batch_by_length
     def bucket_batch_by_length(self, column_names, bucket_boundaries, bucket_batch_sizes,