test: use asynccontextmanager for FastAPI lifespan

opentensor · Jan 22, 2025 · 0958209 · 0958209
1 parent 94749d2
commit 0958209
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit_tests/test_axon.py b/tests/unit_tests/test_axon.py
@@ -16,6 +16,7 @@
 # DEALINGS IN THE SOFTWARE.
 
 
+import contextlib
 import re
 import threading
 import time
@@ -795,6 +796,7 @@ async def test_threaded_fastapi():
     server_started = threading.Event()
     server_stopped = threading.Event()
 
+    @contextlib.asynccontextmanager
     async def lifespan(app):
         server_started.set()
         yield