typo

0a221ab8 · rusty1s · f5cb51ae · 0a221ab8
Commit 0a221ab8 authored Jan 21, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/test_segment.py test/test_segment.py +2 -2

No files found.
--- a/test/test_segment.py
+++ b/test/test_segment.py
@@ -111,8 +111,8 @@ def test_backward(test, reduce, device):
    index = tensor(test['index'], torch.long, device)
    indptr = tensor(test['indptr'], torch.long, device)

-    assert gradcheck(segment_coo, (src, index, None, None, reduce)) is True
-    assert gradcheck(segment_csr, (src, indptr, None, reduce)) is True
+    assert gradcheck(segment_coo, (src, index, None, None, reduce))
+    assert gradcheck(segment_csr, (src, indptr, None, reduce))


 @pytest.mark.parametrize('test,reduce,dtype,device',