Skip to content

Commit 9fbe990

Browse files
committed
Fix the matmul test
1 parent 89d7e5b commit 9fbe990

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_tests/test_linalg.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,7 @@ def test_matmul(x1, x2):
271271
else:
272272
res = _array_module.matmul(x1, x2)
273273

274-
assert res.dtype == dh.promotion_table[x1, x2], "matmul() did not return the correct dtype"
274+
assert res.dtype == dh.promotion_table[x1.dtype, x2.dtype], "matmul() did not return the correct dtype"
275275

276276
if len(x1.shape) == len(x2.shape) == 1:
277277
assert res.shape == ()

0 commit comments

Comments
 (0)