update

Author: tchaton

.lightning/workflows/tests.yml

@@ -52,7 +52,7 @@ run: |
   uv run py-tree /var/tmp/torch
 
   # Testing
-  uv run coverage run --source litserve -m pytest src tests -v
+  uv run coverage run --source litserve -m pytest src tests -v --capture=no
 
   # Coverage reporting
   uv run coverage report

tests/unit/test_lit_server.py

@@ -773,7 +773,7 @@ async def test_worker_restart_and_server_shutdown():
             resp = await ac.post("/predict", json={"input": 0}, timeout=2)
             assert resp.status_code == 500
 
-            await asyncio.sleep(1)
+            await asyncio.sleep(2)
 
             tasks = []
             for _ in range(50):