forgot

run-llama · Nov 28, 2024 · d112c1b · d112c1b
1 parent 0219e5c
commit d112c1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/services/test_workflow_service.py b/tests/services/test_workflow_service.py
@@ -102,7 +102,7 @@ async def test_workflow_service(
     assert result.data["result"] == "test_arg1_result"
 
     # allow a clean shutdown
-    await asyncio.sleep(0)
+    await asyncio.gather(mq_task, server_task, return_exceptions=True)
 
 
 @pytest.mark.asyncio()