Skip to content

Commit 6a1f943

Browse files
authored
Merge pull request #237 from asmeurer/tensordot-fix
Avoid passing a list of axes to tensordot
2 parents 2c12312 + 5484041 commit 6a1f943

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: array_api_tests/test_linalg.py

+1
Original file line numberDiff line numberDiff line change
@@ -768,6 +768,7 @@ def _test_tensordot_stacks(x1, x2, kw, res):
768768
indices = [range(len(s))[i] for i in a]
769769
repl = dict(zip(sorted(indices), range(len(indices))))
770770
res_axes.append(tuple(repl[i] for i in indices))
771+
res_axes = tuple(res_axes)
771772

772773
for ((i,), (j,)), (res_idx,) in zip(
773774
itertools.product(

0 commit comments

Comments
 (0)