diff --git a/python/mxnet/gluon/estimator/__init__.py b/python/mxnet/contrib/estimator/__init__.py similarity index 100% rename from python/mxnet/gluon/estimator/__init__.py rename to python/mxnet/contrib/estimator/__init__.py diff --git a/python/mxnet/gluon/estimator/estimator.py b/python/mxnet/contrib/estimator/estimator.py similarity index 100% rename from python/mxnet/gluon/estimator/estimator.py rename to python/mxnet/contrib/estimator/estimator.py diff --git a/python/mxnet/gluon/estimator/event_handler.py b/python/mxnet/contrib/estimator/event_handler.py similarity index 100% rename from python/mxnet/gluon/estimator/event_handler.py rename to python/mxnet/contrib/estimator/event_handler.py diff --git a/tests/nightly/estimator/test_estimator_cnn.py b/tests/nightly/estimator/test_estimator_cnn.py index b4311b3b3a1e..92d7889ac622 100644 --- a/tests/nightly/estimator/test_estimator_cnn.py +++ b/tests/nightly/estimator/test_estimator_cnn.py @@ -22,7 +22,7 @@ import mxnet as mx from mxnet import gluon, init, nd from mxnet.gluon import data -from mxnet.gluon.estimator import estimator +from mxnet.contrib.estimator import estimator from mxnet.gluon.model_zoo import vision def load_data_mnist(batch_size, resize=None, num_workers=4): diff --git a/tests/nightly/estimator/test_sentiment_rnn.py b/tests/nightly/estimator/test_sentiment_rnn.py index c9dcbd2c8050..89b677852723 100644 --- a/tests/nightly/estimator/test_sentiment_rnn.py +++ b/tests/nightly/estimator/test_sentiment_rnn.py @@ -29,7 +29,7 @@ from mxnet import nd, gluon from mxnet.contrib import text from mxnet.gluon import nn, rnn -from mxnet.gluon.estimator import estimator +from mxnet.contrib.estimator import estimator class TextCNN(nn.Block): diff --git a/tests/python/unittest/test_gluon_estimator.py b/tests/python/unittest/test_gluon_estimator.py index c86f4ff8587e..fd6693d7bd54 100644 --- a/tests/python/unittest/test_gluon_estimator.py +++ b/tests/python/unittest/test_gluon_estimator.py @@ -24,7 +24,7 @@ import mxnet as mx from mxnet import gluon from mxnet.gluon import nn -from mxnet.gluon.estimator import Estimator, EventHandler +from mxnet.contrib.estimator import Estimator, EventHandler from nose.tools import assert_raises diff --git a/tests/python/unittest/test_gluon_event_handler.py b/tests/python/unittest/test_gluon_event_handler.py index 023b04691a39..f3181f5e4f23 100644 --- a/tests/python/unittest/test_gluon_event_handler.py +++ b/tests/python/unittest/test_gluon_event_handler.py @@ -20,7 +20,7 @@ import mxnet as mx from mxnet import nd from mxnet.gluon import nn, loss -from mxnet.gluon.estimator import estimator, event_handler +from mxnet.contrib.estimator import estimator, event_handler def _get_test_network(): net = nn.Sequential()