We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 2c12312 + 5484041 commit 6a1f943Copy full SHA for 6a1f943
1 file changed
array_api_tests/test_linalg.py
@@ -768,6 +768,7 @@ def _test_tensordot_stacks(x1, x2, kw, res):
768
indices = [range(len(s))[i] for i in a]
769
repl = dict(zip(sorted(indices), range(len(indices))))
770
res_axes.append(tuple(repl[i] for i in indices))
771
+ res_axes = tuple(res_axes)
772
773
for ((i,), (j,)), (res_idx,) in zip(
774
itertools.product(
0 commit comments