diff --git a/tests/test_dependency_after_yield_sse.py b/tests/test_dependency_after_yield_sse.py index e9599380c8..e600f565d0 100644 --- a/tests/test_dependency_after_yield_sse.py +++ b/tests/test_dependency_after_yield_sse.py @@ -30,9 +30,8 @@ class Session: async def _async_iter(self) -> AsyncGenerator[Item, None]: for item in self.items: - if self.open: - yield item - else: + yield item + if not self.open: raise ValueError("Session closed")