mv graph_map_schema.py to example

example/graph_to_mindrecord/writer.py

@@ -24,7 +24,7 @@ from importlib import import_module
 from multiprocessing import Pool
 
 from mindspore.mindrecord import FileWriter
-from mindspore.mindrecord import GraphMapSchema
+from graph_map_schema import GraphMapSchema
 
 
 def exec_task(task_id, parallel_writer=True):

mindspore/mindrecord/__init__.inter

@@ -32,7 +32,6 @@ from .mindpage import MindPage
 from .shardutils import SUCCESS, FAILED
 from .tools.cifar10_to_mr import Cifar10ToMR
 from .tools.cifar100_to_mr import Cifar100ToMR
-from .tools.graph_map_schema import GraphMapSchema
 from .tools.imagenet_to_mr import ImageNetToMR
 from .tools.mnist_to_mr import MnistToMR
 

mindspore/mindrecord/__init__.py

@@ -29,10 +29,9 @@ from .common.exceptions import *
 from .shardutils import SUCCESS, FAILED
 from .tools.cifar10_to_mr import Cifar10ToMR
 from .tools.cifar100_to_mr import Cifar100ToMR
-from .tools.graph_map_schema import GraphMapSchema
 from .tools.imagenet_to_mr import ImageNetToMR
 from .tools.mnist_to_mr import MnistToMR
 
-__all__ = ['FileWriter', 'FileReader', 'MindPage', 'GraphMapSchema',
+__all__ = ['FileWriter', 'FileReader', 'MindPage',
            'Cifar10ToMR', 'Cifar100ToMR', 'ImageNetToMR', 'MnistToMR',
            'SUCCESS', 'FAILED']