Fixed tests.

spcl · Sep 27, 2023 · 6490b2f · 6490b2f
1 parent e34ca3c
commit 6490b2f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/sdfg/data/structure_test.py b/tests/sdfg/data/structure_test.py
@@ -466,9 +466,7 @@ def test_direct_read_structure_loops():
     state.add_edge(t, '__out', B, None, dace.Memlet(data='B', subset='0:M, 0:N', volume=1))
 
     idx_before, idx_guard, idx_after = sdfg.add_loop(None, state, None, 'idx', 'A.indptr[i]', 'idx < A.indptr[i+1]', 'idx + 1')
-    i_before, i_guard, i_after = sdfg.add_loop(None, idx_before, None, 'i', '0', 'i < M', 'i + 1')
-
-    sdfg.view()
+    i_before, i_guard, i_after = sdfg.add_loop(None, idx_before, None, 'i', '0', 'i < M', 'i + 1', loop_end_state=idx_after)
 
     func = sdfg.compile()