BUG: fix triu_indices

Author: ev-br

torch_np/_wrapper.py

@@ -772,7 +772,7 @@ def tril_indices(n, k=0, m=None):
 def triu_indices(n, k=0, m=None):
     if m is None:
         m = n
-    tensor_2 = torch.tril_indices(n, m, offset=k)
+    tensor_2 = torch.triu_indices(n, m, offset=k)
     return tuple(asarray(_) for _ in tensor_2)
 
 

torch_np/tests/numpy_tests/lib/test_twodim_base.py

@@ -365,7 +365,6 @@ def test_mask_indices():
     assert_array_equal(a[iu1], array([1, 2, 5]))
 
 
-@pytest.mark.xfail(reason="TODO: fancy indexing")
 def test_tril_indices():
     # indices without and with offset
     il1 = tril_indices(4)
@@ -413,7 +412,6 @@ def test_tril_indices():
                               [-10, -10, -10, -10, -10]]))
 
 
-@pytest.mark.xfail(reason="TODO: fancy indexing")
 class TestTriuIndices:
     def test_triu_indices(self):
         iu1 = triu_indices(4)