From 972f727a7534d432f95964669873591b817aa86b Mon Sep 17 00:00:00 2001 From: heleiwang Date: Wed, 27 May 2020 16:43:32 +0800 Subject: [PATCH] mv graph_map_schema.py to example --- .../tools => example/graph_to_mindrecord}/graph_map_schema.py | 0 example/graph_to_mindrecord/writer.py | 2 +- mindspore/mindrecord/__init__.inter | 1 - mindspore/mindrecord/__init__.py | 3 +-- 4 files changed, 2 insertions(+), 4 deletions(-) rename {mindspore/mindrecord/tools => example/graph_to_mindrecord}/graph_map_schema.py (100%) diff --git a/mindspore/mindrecord/tools/graph_map_schema.py b/example/graph_to_mindrecord/graph_map_schema.py similarity index 100% rename from mindspore/mindrecord/tools/graph_map_schema.py rename to example/graph_to_mindrecord/graph_map_schema.py diff --git a/example/graph_to_mindrecord/writer.py b/example/graph_to_mindrecord/writer.py index 4ada8d60ea..1024c82372 100644 --- 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): diff --git a/mindspore/mindrecord/__init__.inter b/mindspore/mindrecord/__init__.inter index 6fcabc2c5f..ca1d50153b 100644 --- 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 diff --git a/mindspore/mindrecord/__init__.py b/mindspore/mindrecord/__init__.py index 44fc990399..31fb801c46 100644 --- 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']