diff --git a/tests/test_dependency_after_yield_raise.py b/tests/test_dependency_after_yield_raise.py index b56140277..b560dc36f 100644 --- a/tests/test_dependency_after_yield_raise.py +++ b/tests/test_dependency_after_yield_raise.py @@ -1,8 +1,9 @@ -from typing import Annotated, Any +from typing import Any import pytest from fastapi import Depends, FastAPI, HTTPException from fastapi.testclient import TestClient +from typing_extensions import Annotated class CustomError(Exception): diff --git a/tests/test_dependency_after_yield_streaming.py b/tests/test_dependency_after_yield_streaming.py index 5235d5249..7e1c8822b 100644 --- a/tests/test_dependency_after_yield_streaming.py +++ b/tests/test_dependency_after_yield_streaming.py @@ -1,10 +1,11 @@ from contextlib import contextmanager -from typing import Annotated, Any, Generator +from typing import Any, Generator import pytest from fastapi import Depends, FastAPI from fastapi.responses import StreamingResponse from fastapi.testclient import TestClient +from typing_extensions import Annotated class Session: diff --git a/tests/test_dependency_after_yield_websockets.py b/tests/test_dependency_after_yield_websockets.py index 3fa1b051b..7c323c338 100644 --- a/tests/test_dependency_after_yield_websockets.py +++ b/tests/test_dependency_after_yield_websockets.py @@ -1,9 +1,10 @@ from contextlib import contextmanager -from typing import Annotated, Any, Generator +from typing import Any, Generator import pytest from fastapi import Depends, FastAPI, WebSocket from fastapi.testclient import TestClient +from typing_extensions import Annotated class Session: