sanity

apache · Mar 11, 2020 · 40c688c · 40c688c
1 parent 76aa9ce
commit 40c688c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/python/mxnet/ndarray/numpy/_op.py b/python/mxnet/ndarray/numpy/_op.py
@@ -6789,9 +6789,7 @@ def median(a, axis=None, out=None, overwrite_input=None, keepdims=False):
     >>> np.median(a, axis=1)
     array([7.,  2.])
     """
-    from mxnet import np, npx
-    npx.set_np()
-    return quantile(a=a, q=np.array(0.5), axis=axis, out=out, overwrite_input=overwrite_input,
+    return quantile(a=a, q=0.5, axis=axis, out=out, overwrite_input=overwrite_input,
                     interpolation='midpoint', keepdims=keepdims)
 
 

diff --git a/python/mxnet/numpy/multiarray.py b/python/mxnet/numpy/multiarray.py
@@ -8812,7 +8812,7 @@ def median(a, axis=None, out=None, overwrite_input=None, keepdims=False):
     array([7.,  2.])
     """
     return _mx_nd_np.median(a, axis=axis, overwrite_input=overwrite_input,
-           keepdims=keepdims, out=out)
+                            keepdims=keepdims, out=out)
 
 
 @set_module('mxnet.numpy')

diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py
@@ -7284,7 +7284,6 @@ def hybrid_forward(self, F, a):
         assert_almost_equal(mx_out.asnumpy(), np_out, atol=atol, rtol=rtol)
 
 
-
 @with_seed()
 @use_np
 def test_np_quantile():