mv graph_map_schema.py to example

5 years ago · 972f727a75
parent 9c20b1d6f7
commit 972f727a75
4 changed files with 2 additions and 4 deletions
--- a/example/graph_to_mindrecord/graph_map_schema.py
+++ b/example/graph_to_mindrecord/graph_map_schema.py
--- a/example/graph_to_mindrecord/writer.py
+++ b/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):
--- a/mindspore/mindrecord/init.inter
+++ b/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

--- a/mindspore/mindrecord/init.py
+++ b/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']