diff --git a/tests/test_dependency_types.py b/tests/test_dependency_types.py index d4bf99061..dd1741421 100644 --- a/tests/test_dependency_types.py +++ b/tests/test_dependency_types.py @@ -16,12 +16,12 @@ async def async_function_dependency(value: str) -> str: return value -def gen_dependency(value: str) -> str: +def gen_dependency(value: str) -> Generator[str, None, None]: yield value -async def async_gen_dependency(value: str) -> str: - return value +async def async_gen_dependency(value: str) -> AsyncGenerator[str, None]: + yield value class CallableDependency: