fix GeneratorDataset unable to do serialize

2021-01-30 11:32:40 +08:00 · 2021-01-30 11:32:40 +08:00 · fa2a854057
parent 84afbb510f
commit fa2a854057
1 changed files with 2 additions and 1 deletions
--- 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,