Fuller type for Generator

Author: neoncube2

playwright/_impl/_connection.py

@@ -513,7 +513,7 @@ def _replace_guids_with_channels(self, payload: Any) -> Any:
     def needs_full_stack_trace(self) -> bool:
         return self._tracing_count > 0
 
-    def get_frame_info(self) -> Generator[FrameInfo]:
+    def get_frame_info(self) -> Generator[FrameInfo, None]:
         current_frame = inspect.currentframe()
 
         if current_frame is None:
@@ -542,7 +542,7 @@ async def wrap_api_call(
         if self._api_zone.get():
             return await cb()
         task = asyncio.current_task(self._loop)
-        st: Union[List[FrameInfo], Generator[FrameInfo]] = (
+        st: Union[List[FrameInfo], Generator[FrameInfo, None]] = (
             getattr(task, "__pw_stack__", None) or self.get_frame_info()
         )
         parsed_st = _extract_stack_trace_information_from_stack(
@@ -562,7 +562,7 @@ def wrap_api_call_sync(
         if self._api_zone.get():
             return cb()
         task = asyncio.current_task(self._loop)
-        st: Union[List[FrameInfo], Generator[FrameInfo]] = (
+        st: Union[List[FrameInfo], Generator[FrameInfo, None]] = (
             getattr(task, "__pw_stack__", None) or self.get_frame_info()
         )
         parsed_st = _extract_stack_trace_information_from_stack(
@@ -598,7 +598,7 @@ class ParsedStackTrace(TypedDict):
 
 
 def _extract_stack_trace_information_from_stack(
-    st: Union[List[FrameInfo], Generator[FrameInfo]],
+    st: Union[List[FrameInfo], Generator[FrameInfo, None]],
     is_internal: bool,
     needs_full_stack: bool,
 ) -> ParsedStackTrace: