fix __aiter__ return type (#223)

Author: theomonnom

livekit-rtc/livekit/rtc/audio_stream.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import asyncio
-from typing import Optional
+from typing import Optional, AsyncIterator
 
 from ._ffi_client import FfiHandle, FfiClient
 from ._proto import audio_frame_pb2 as proto_audio_frame
@@ -78,12 +78,12 @@ async def aclose(self) -> None:
         self._ffi_handle.dispose()
         await self._task
 
-    def __aiter__(self) -> "AudioStream":
-        return self
-
     def _is_event(self, e: proto_ffi.FfiEvent) -> bool:
         return e.audio_stream_event.stream_handle == self._ffi_handle.handle
 
+    def __aiter__(self) -> AsyncIterator[AudioFrameEvent]:
+        return self
+
     async def __anext__(self) -> AudioFrameEvent:
         if self._task.done():
             raise StopAsyncIteration

livekit-rtc/livekit/rtc/video_stream.py

@@ -14,7 +14,7 @@
 
 import asyncio
 from dataclasses import dataclass
-from typing import Optional
+from typing import Optional, AsyncIterator
 
 from ._ffi_client import FfiHandle, FfiClient
 from ._proto import ffi_pb2 as proto_ffi
@@ -91,12 +91,12 @@ async def aclose(self) -> None:
         self._ffi_handle.dispose()
         await self._task
 
-    def __aiter__(self) -> "VideoStream":
-        return self
-
     def _is_event(self, e: proto_ffi.FfiEvent) -> bool:
         return e.video_stream_event.stream_handle == self._ffi_handle.handle
 
+    def __aiter__(self) -> AsyncIterator[VideoFrameEvent]:
+        return self
+
     async def __anext__(self) -> VideoFrameEvent:
         if self._task.done():
             raise StopAsyncIteration