diff --git a/python/mxnet/numpy/multiarray.py b/python/mxnet/numpy/multiarray.py index 4910b4d6b925..a1b0e016445a 100644 --- a/python/mxnet/numpy/multiarray.py +++ b/python/mxnet/numpy/multiarray.py @@ -1377,7 +1377,7 @@ def argsort(self, axis=-1, kind=None, order=None): # pylint: disable=arguments- The arguments are the same as for :py:func:`argsort`, with this array as data. """ - raise argsort(self, axis=axis, kind=kind, order=order) + return argsort(self, axis=axis, kind=kind, order=order) def argmax_channel(self, *args, **kwargs): """Convenience fluent method for :py:func:`argmax_channel`. diff --git a/python/mxnet/symbol/numpy/_symbol.py b/python/mxnet/symbol/numpy/_symbol.py index 6efc333cc16c..3ee385660715 100644 --- a/python/mxnet/symbol/numpy/_symbol.py +++ b/python/mxnet/symbol/numpy/_symbol.py @@ -491,7 +491,7 @@ def argsort(self, axis=-1, kind=None, order=None): # pylint: disable=arguments- The arguments are the same as for :py:func:`argsort`, with this array as data. """ - raise argsort(self, axis=axis, kind=kind, order=order) + return argsort(self, axis=axis, kind=kind, order=order) def argmax_channel(self, *args, **kwargs): """Convenience fluent method for :py:func:`argmax_channel`.