diff --git a/python/mxnet/ndarray/numpy/random.py b/python/mxnet/ndarray/numpy/random.py index f38e5bb53f8a..8607fd5c3514 100644 --- a/python/mxnet/ndarray/numpy/random.py +++ b/python/mxnet/ndarray/numpy/random.py @@ -23,7 +23,6 @@ from ..ndarray import NDArray from . import _internal as _npi - __all__ = ['uniform', 'normal', 'multinomial'] diff --git a/python/mxnet/numpy/random.py b/python/mxnet/numpy/random.py index 7ef59adb52d8..cda1adafe24f 100644 --- a/python/mxnet/numpy/random.py +++ b/python/mxnet/numpy/random.py @@ -99,6 +99,7 @@ def normal(loc=0.0, scale=1.0, size=None, **kwargs): """ return _mx_nd_np.random.normal(loc, scale, size, **kwargs) + def multinomial(n, pvals, size=None, **kwargs): """Draw samples from a multinomial distribution. diff --git a/tests/python/unittest/test_numpy_ndarray.py b/tests/python/unittest/test_numpy_ndarray.py index cdbdd0abd111..a158468cb7bd 100644 --- a/tests/python/unittest/test_numpy_ndarray.py +++ b/tests/python/unittest/test_numpy_ndarray.py @@ -670,7 +670,6 @@ def test_np_multinomial(): assert freq.size == 0 - if __name__ == '__main__': import nose nose.runmodule()