Skip to content

Commit f494b45

Browse files
committed
Don't compare float elements in test_tensordot
1 parent 9bccfa5 commit f494b45

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

array_api_tests/test_linalg.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -686,7 +686,7 @@ def _test_tensordot_stacks(x1, x2, kw, res):
686686
x1_stack = x1[i]
687687
x2_stack = x2[j]
688688
decomp_res_stack = xp.tensordot(x1_stack, x2_stack, axes=res_axes)
689-
assert_exactly_equal(res_stack, decomp_res_stack)
689+
assert_equal(res_stack, decomp_res_stack)
690690

691691
@given(
692692
*two_mutual_arrays(dh.numeric_dtypes, two_shapes=tensordot_shapes()),
@@ -771,7 +771,6 @@ def test_vecdot(dtypes, shape, data):
771771
ph.assert_dtype("vecdot", dtypes, out.dtype)
772772
# TODO: assert shape and elements
773773

774-
775774
# Insanely large orders might not work. There isn't a limit specified in the
776775
# spec, so we just limit to reasonable values here.
777776
max_ord = 100

0 commit comments

Comments
 (0)