Fix

taichi-dev · Nov 18, 2021 · a05f04d · a05f04d
1 parent 0996cf7
commit a05f04d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/python/test_ast_refactor.py b/tests/python/test_ast_refactor.py
@@ -849,8 +849,7 @@ def test_listcomp():
     @ti.func
     def identity(dt, n: ti.template()):
         return ti.Matrix([[ti.cast(int(i == j), dt) for j in range(n)]
-                          for i in range(n)],
-                         disable_local_tensor=1)
+                          for i in range(n)])
 
     @ti.kernel
     def foo(n: ti.template()) -> ti.i32: