Skip to content

Commit 1daba5d

Browse files
committed
Fix the true_value check for test_vecdot
1 parent 72974e0 commit 1daba5d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

array_api_tests/test_linalg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _test_stacks(f, *args, res=None, dims=2, true_val=None,
9797
msg_extra = f'{x_idxes = }, {res_idx = }'
9898
assert_equal(res_stack, decomp_res_stack, msg_extra)
9999
if true_val:
100-
assert_equal(decomp_res_stack, true_val(*x_stacks), msg_extra)
100+
assert_equal(decomp_res_stack, true_val(*x_stacks, **kw), msg_extra)
101101

102102
def _test_namedtuple(res, fields, func_name):
103103
"""
@@ -789,7 +789,7 @@ def test_vecdot(x1, x2, kw):
789789
ph.assert_shape("vecdot", res.shape, expected_shape)
790790

791791
if x1.dtype in dh.int_dtypes:
792-
def true_val(x, y, axix=-1):
792+
def true_val(x, y, axis=-1):
793793
return xp.sum(x*y, dtype=res.dtype)
794794
else:
795795
true_val = None

0 commit comments

Comments
 (0)