From 57791d82ebd22c2e5359185e05f18fb46d662f96 Mon Sep 17 00:00:00 2001
From: bai-yangfan <baiyangfan@huawei.com>
Date: Mon, 14 Sep 2020 12:44:58 +0800
Subject: [PATCH] all_notes

---
 mindspore/common/__init__.py | 8 ++++----
 mindspore/nn/__init__.py     | 2 ++
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/mindspore/common/__init__.py b/mindspore/common/__init__.py
index 2ede99a603..64f7d03e02 100644
--- a/mindspore/common/__init__.py
+++ b/mindspore/common/__init__.py
@@ -20,12 +20,12 @@ from .parameter import Parameter, ParameterTuple
 from .tensor import MetaTensor, Tensor, RowTensor, SparseTensor
 from .seed import set_seed, get_seed
 
-__all__ = [
+
+__all__ = dtype.__all__
+__all__.extend([
     "MetaTensor", "Tensor", "RowTensor", "SparseTensor",  # tensor
     'ms_function',  # api
     'Parameter', 'ParameterTuple',  # parameter
     "dtype",
     "set_seed", "get_seed"  # random seed
-    ]
-
-__all__.extend(dtype.__all__)
+    ])
diff --git a/mindspore/nn/__init__.py b/mindspore/nn/__init__.py
index 7f69bc86d1..cee7b08915 100644
--- a/mindspore/nn/__init__.py
+++ b/mindspore/nn/__init__.py
@@ -18,6 +18,7 @@ Neural Networks Cells.
 Pre-defined building blocks or computing units to construct Neural Networks.
 """
 from . import layer, loss, optim, metrics, wrap, probability, sparse
+from .learning_rate_schedule import *
 from .cell import Cell, GraphKernel
 from .layer import *
 from .loss import *
@@ -34,5 +35,6 @@ __all__.extend(optim.__all__)
 __all__.extend(metrics.__all__)
 __all__.extend(wrap.__all__)
 __all__.extend(sparse.__all__)
+__all__.extend(learning_rate_schedule.__all__)
 
 __all__.sort()