26
26
Any ,
27
27
Callable ,
28
28
Dict ,
29
- Generator ,
29
+ Iterator ,
30
30
List ,
31
31
Mapping ,
32
32
Optional ,
@@ -513,7 +513,7 @@ def _replace_guids_with_channels(self, payload: Any) -> Any:
513
513
def needs_full_stack_trace (self ) -> bool :
514
514
return self ._tracing_count > 0
515
515
516
- def get_frame_info (self ) -> Generator [FrameInfo , None ]:
516
+ def get_frame_info (self ) -> Iterator [FrameInfo ]:
517
517
current_frame = inspect .currentframe ()
518
518
519
519
if current_frame is None :
@@ -542,7 +542,7 @@ async def wrap_api_call(
542
542
if self ._api_zone .get ():
543
543
return await cb ()
544
544
task = asyncio .current_task (self ._loop )
545
- st : Union [List [FrameInfo ], Generator [FrameInfo , None ]] = (
545
+ st : Union [List [FrameInfo ], Iterator [FrameInfo ]] = (
546
546
getattr (task , "__pw_stack__" , None ) or self .get_frame_info ()
547
547
)
548
548
parsed_st = _extract_stack_trace_information_from_stack (
@@ -562,7 +562,7 @@ def wrap_api_call_sync(
562
562
if self ._api_zone .get ():
563
563
return cb ()
564
564
task = asyncio .current_task (self ._loop )
565
- st : Union [List [FrameInfo ], Generator [FrameInfo , None ]] = (
565
+ st : Union [List [FrameInfo ], Iterator [FrameInfo ]] = (
566
566
getattr (task , "__pw_stack__" , None ) or self .get_frame_info ()
567
567
)
568
568
parsed_st = _extract_stack_trace_information_from_stack (
@@ -598,7 +598,7 @@ class ParsedStackTrace(TypedDict):
598
598
599
599
600
600
def _extract_stack_trace_information_from_stack (
601
- st : Union [List [FrameInfo ], Generator [FrameInfo , None ]],
601
+ st : Union [List [FrameInfo ], Iterator [FrameInfo ]],
602
602
is_internal : bool ,
603
603
needs_full_stack : bool ,
604
604
) -> ParsedStackTrace :
0 commit comments