numpy.msort no longer exists

soundradix · Jul 9, 2024 · d2be119 · d2be119
1 parent 2517224
commit d2be119
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 3 deletions.
diff --git a/autograd/numpy/numpy_jvps.py b/autograd/numpy/numpy_jvps.py
@@ -210,7 +210,6 @@ def fwd_grad_sort(g, ans, x, axis=-1, kind='quicksort', order=None):
     sort_perm = anp.argsort(x, axis, kind, order)
     return g[sort_perm]
 defjvp(anp.sort, fwd_grad_sort)
-defjvp(anp.msort, lambda g, ans, x: fwd_grad_sort(g, ans, x, axis=0))
 
 def fwd_grad_partition(g, ans, x, kth, axis=-1, kind='introselect', order=None):
     partition_perm = anp.argpartition(x, kth, axis, kind, order)

diff --git a/autograd/numpy/numpy_vjps.py b/autograd/numpy/numpy_vjps.py
@@ -560,7 +560,6 @@ def grad_sort(ans, x, axis=-1, kind='quicksort', order=None):
     sort_perm = anp.argsort(x, axis, kind, order)
     return lambda g: unpermuter(g, sort_perm)
 defvjp(anp.sort, grad_sort)
-defvjp(anp.msort, grad_sort)  # Until multi-D is allowed, these are the same.
 
 def grad_partition(ans, x, kth, axis=-1, kind='introselect', order=None):
     #TODO: Cast input with np.asanyarray()

diff --git a/tests/test_systematic.py b/tests/test_systematic.py
@@ -155,7 +155,6 @@ def test_fmin(): combo_check(np.fmin, [0, 1])(
                             [R(1), R(1,4), R(3, 4)])
 
 def test_sort():       combo_check(np.sort, [0])([R(1), R(7)])
-def test_msort():     combo_check(np.msort, [0])([R(1), R(7)])
 def test_partition(): combo_check(np.partition, [0])(
                                   [R(7), R(14)], kth=[0, 3, 6])