Merge pull request #237 from asmeurer/tensordot-fix

Avoid passing a list of axes to tensordot
data-apis · Feb 28, 2024 · 6a1f943 · 6a1f943
2 parents 2c12312 + 5484041
commit 6a1f943
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/array_api_tests/test_linalg.py b/array_api_tests/test_linalg.py
@@ -768,6 +768,7 @@ def _test_tensordot_stacks(x1, x2, kw, res):
             indices = [range(len(s))[i] for i in a]
             repl = dict(zip(sorted(indices), range(len(indices))))
             res_axes.append(tuple(repl[i] for i in indices))
+        res_axes = tuple(res_axes)
 
     for ((i,), (j,)), (res_idx,) in zip(
             itertools.product(