better wait, log only necessary things

key4hep · Jul 11, 2024 · ca020c2 · ca020c2
1 parent 22af9ed
commit ca020c2
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/test/scheduling.jl b/test/scheduling.jl
@@ -41,18 +41,12 @@ end
     algorithms_count = ilength(MetaGraphs.filter_vertices(graph, :type, "Algorithm"))
     set_indexing_prop!(graph, :node_id)
 
-    Dagger.enable_logging!(timeline=true,
-        tasknames=true,
-        taskdeps=true,
-        taskargs=true,
-        taskargmoves=true,
-    )
+    Dagger.enable_logging!(tasknames=true, taskdeps=true)
     _ = Dagger.fetch_logs!() # flush logs
 
-    FrameworkDemo.schedule_graph(graph)
-    for v in vertices(graph)
-        wait(get_prop(graph, v, :res_data))
-    end
+    tasks = FrameworkDemo.schedule_graph(graph)
+    wait.(tasks)
+
     logs = Dagger.fetch_logs!()
     @test !isnothing(logs)