From ca020c213dd16d20684a380b426e343dade554a2 Mon Sep 17 00:00:00 2001
From: Mateusz Jakub Fila <mateusz.jakub.fila@cern.ch>
Date: Thu, 11 Jul 2024 17:38:52 +0200
Subject: [PATCH] better wait, log only necessary things

---
 test/scheduling.jl | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/test/scheduling.jl b/test/scheduling.jl
index 6d31252..7d74463 100644
--- 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)