style: annotate return type

winstxnhdw · Feb 27, 2024 · 5e2e63d · 5e2e63d
1 parent 59b4a0c
commit 5e2e63d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_translate.py b/tests/test_translate.py
@@ -1,13 +1,15 @@
 # pylint: disable=missing-function-docstring,redefined-outer-name
 
+from typing import Generator
+
 from fastapi.testclient import TestClient
 from pytest import fixture
 
 from server import initialise
 
 
 @fixture()
-def client():
+def client() -> Generator[TestClient, None, None]:
     with TestClient(initialise()) as client:
         yield client