Skip to content

Commit 7063c6f

Browse files
committed
fix test_backends.py
1 parent efaee05 commit 7063c6f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_backends.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1194,12 +1194,12 @@ def test_backend_all(backend):
11941194
# Test all True
11951195
a = tc.backend.convert_to_tensor([True, True, True])
11961196
result = tc.backend.all(a)
1197-
assert result is True
1197+
assert tc.backend.numpy(result).item() is True
11981198

11991199
# Test with False
12001200
b = tc.backend.convert_to_tensor([True, False, True])
12011201
result = tc.backend.all(b)
1202-
assert result is False
1202+
assert tc.backend.numpy(result).item() is False
12031203

12041204

12051205
@pytest.mark.parametrize("backend", [lf("npb"), lf("tfb"), lf("jaxb"), lf("torchb")])

0 commit comments

Comments
 (0)