fix return type errors

Author: lukasIO

livekit-rtc/livekit/rtc/audio_frame.py

@@ -186,7 +186,7 @@ def __repr__(self) -> str:
         )
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, *_: Any):
+    def __get_pydantic_core_schema__(cls, *_: Any) -> Any:
         from pydantic_core import core_schema
         import base64
 

livekit-rtc/livekit/rtc/event_emitter.py

@@ -1,6 +1,6 @@
 import inspect
 import asyncio
-from typing import Callable, Dict, Set, Optional, Generic, TypeVar
+from typing import Any, Callable, Dict, Set, Optional, Generic, TypeVar
 
 from .log import logger
 
@@ -14,7 +14,7 @@ def __init__(self) -> None:
         """
         self._events: Dict[T_contra, Set[Callable]] = dict()
 
-    def emit(self, event: T_contra, *args) -> None:
+    def emit(self, event: T_contra, *args: Any) -> None:
         """
         Trigger all callbacks associated with the given event.
 
@@ -104,7 +104,7 @@ def greet_once(name):
         """
         if callback is not None:
 
-            def once_callback(*args, **kwargs):
+            def once_callback(*args: Any, **kwargs: Any) -> None:
                 self.off(event, once_callback)
                 callback(*args, **kwargs)
 

livekit-rtc/livekit/rtc/synchronizer.py

@@ -2,7 +2,7 @@
 import logging
 import time
 from collections import deque
-from typing import Optional, Union
+from typing import Any, Optional, Union
 
 from .video_frame import VideoFrame
 from .audio_frame import AudioFrame
@@ -149,7 +149,7 @@ def __init__(self, *, expected_fps: float, max_delay_tolerance_ms: float = 300)
     async def __aenter__(self) -> None:
         await self.wait_next_process()
 
-    async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
+    async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
         self.after_process()
 
     def reset(self) -> None:

livekit-rtc/livekit/rtc/track.py

@@ -80,14 +80,14 @@ def create_audio_track(name: str, source: "AudioSource") -> "LocalAudioTrack":
         resp = FfiClient.instance.request(req)
         return LocalAudioTrack(resp.create_audio_track.track)
 
-    def mute(self):
+    def mute(self) -> None:
         req = proto_ffi.FfiRequest()
         req.local_track_mute.track_handle = self._ffi_handle.handle
         req.local_track_mute.mute = True
         FfiClient.instance.request(req)
         self._info.muted = True
 
-    def unmute(self):
+    def unmute(self) -> None:
         req = proto_ffi.FfiRequest()
         req.local_track_mute.track_handle = self._ffi_handle.handle
         req.local_track_mute.mute = False
@@ -111,14 +111,14 @@ def create_video_track(name: str, source: "VideoSource") -> "LocalVideoTrack":
         resp = FfiClient.instance.request(req)
         return LocalVideoTrack(resp.create_video_track.track)
 
-    def mute(self):
+    def mute(self) -> None:
         req = proto_ffi.FfiRequest()
         req.local_track_mute.track_handle = self._ffi_handle.handle
         req.local_track_mute.mute = True
         FfiClient.instance.request(req)
         self._info.muted = True
 
-    def unmute(self):
+    def unmute(self) -> None:
         req = proto_ffi.FfiRequest()
         req.local_track_mute.track_handle = self._ffi_handle.handle
         req.local_track_mute.mute = False

livekit-rtc/livekit/rtc/track_publication.py

@@ -106,7 +106,7 @@ def track(self) -> Optional[RemoteTrack]:
     def subscribed(self) -> bool:
         return self._subscribed
 
-    def set_subscribed(self, subscribed: bool):
+    def set_subscribed(self, subscribed: bool) -> None:
         req = proto_ffi.FfiRequest()
         req.set_subscribed.subscribe = subscribed
         req.set_subscribed.publication_handle = self._ffi_handle.handle

livekit-rtc/livekit/rtc/video_frame.py

@@ -202,7 +202,7 @@ def __repr__(self) -> str:
         return f"rtc.VideoFrame(width={self.width}, height={self.height}, type={proto_video.VideoBufferType.Name(self.type)})"
 
     @classmethod
-    def __get_pydantic_core_schema__(cls, *_: Any):
+    def __get_pydantic_core_schema__(cls, *_: Any) -> Any:
         from pydantic_core import core_schema
         import base64
 

livekit-rtc/livekit/rtc/video_stream.py

@@ -44,7 +44,7 @@ def __init__(
         loop: Optional[asyncio.AbstractEventLoop] = None,
         capacity: int = 0,
         format: Optional[proto_video_frame.VideoBufferType.ValueType] = None,
-        **kwargs,
+        **kwargs: Any,
     ) -> None:
         self._loop = loop or asyncio.get_event_loop()
         self._ffi_queue = FfiClient.instance.queue.subscribe(self._loop)