diff --git a/python/mxnet/optimizer.py b/python/mxnet/optimizer.py index 3073893de4a6..e7b8013952b8 100644 --- a/python/mxnet/optimizer.py +++ b/python/mxnet/optimizer.py @@ -2,10 +2,10 @@ import math import pickle import logging +import mxnet as mx from .ndarray import NDArray, clip, sqrt, sign from .ndarray import sgd_update, sgd_mom_update, adam_update, rmsprop_update, rmspropalex_update from .random import normal -import mxnet as mx class Optimizer(object): diff --git a/python/mxnet/test_utils.py b/python/mxnet/test_utils.py index bf4c5b9580c0..38c79aedfcbd 100644 --- a/python/mxnet/test_utils.py +++ b/python/mxnet/test_utils.py @@ -19,10 +19,10 @@ except ImportError: # in rare cases requests may be not installed pass +import mxnet as mx from .context import Context from .ndarray import array, _STORAGE_TYPE_STR_TO_ID from .symbol import Symbol -import mxnet as mx _rng = np.random.RandomState(1234)