fix test_graph.py

taichi-dev · Nov 20, 2023 · 3346e15 · 3346e15
1 parent 57e7648
commit 3346e15
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/tests/python/test_graph.py b/tests/python/test_graph.py
@@ -373,16 +373,13 @@ def paint(
     _rw_tex = ti.graph.Arg(
         ti.graph.ArgKind.RWTEXTURE,
         "rw_tex",
-        channel_format=ti.f32,
-        shape=(128, 128),
-        num_channels=1,
+        ndim=2,
+        fmt=ti.Format.r32f,
     )
     _tex = ti.graph.Arg(
         ti.graph.ArgKind.TEXTURE,
         "tex",
-        channel_format=ti.f32,
-        shape=(128, 128),
-        num_channels=1,
+        ndim=2,
     )
 
     g_builder = ti.graph.GraphBuilder()
@@ -452,9 +449,8 @@ def read(tex: ti.types.texture(num_dimensions=2), arr: ti.types.ndarray()):
     sym_tex = ti.graph.Arg(
         ti.graph.ArgKind.RWTEXTURE,
         "tex",
-        channel_format=ti.f32,
-        shape=res,
-        num_channels=1,
+        fmt=ti.Format.r32f,
+        ndim=2,
     )
     sym_arr = ti.graph.Arg(ti.graph.ArgKind.NDARRAY, "arr", ti.f32, ndim=2)