merge with main branch

taichi-dev · Nov 11, 2022 · 168fc5e · 168fc5e
1 parent aa012ce
commit 168fc5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/taichi/lang/matrix_ops.py b/python/taichi/lang/matrix_ops.py
@@ -79,7 +79,7 @@ def _rotation2d_matrix(alpha):
 
 
 @preconditions(
-    arg_at(0, same_shapes),
+    arg_at(0, lambda xs: same_shapes(*xs)),
     arg_foreach_check(
         0,
         fns=[assert_vector(), assert_list],