fix some broken types

2 files changed, 6 insertions, 8 deletions

diff --git a/backend/decky_loader/plugin/sandboxed_plugin.py b/backend/decky_loader/plugin/sandboxed_plugin.py
index f83636f1..3fd38e4f 100644
--- a/backend/decky_loader/plugin/sandboxed_plugin.py
+++ b/backend/decky_loader/plugin/sandboxed_plugin.py
@@ -83,15 +83,15 @@ class SandboxedPlugin:
                 sysmodules[key.replace("decky_loader.", "")] = sysmodules[key]
             
             from .imports import decky
-            async def emit_message(event: str, data: DataType | None = None, data_type: Type[DataType] | None = None) -> None:
+            async def emit(event: str, data: DataType | None = None, data_type: Type[DataType] | None = None) -> None:
                 await self._socket.write_single_line_server(dumps({
                     "type": SocketMessageType.EVENT,
                     "event": event,
                     "data": data
                 }))
             # copy the docstring over so we don't have to duplicate it
-            emit_message.__doc__ = decky.emit_message.__doc__
-            decky.emit_message = emit_message
+            emit.__doc__ = decky.emit.__doc__
+            decky.emit = emit
             sysmodules["decky"] = decky
             # provided for compatibility
             sysmodules["decky_plugin"] = decky
diff --git a/backend/decky_loader/wsrouter.py b/backend/decky_loader/wsrouter.py
index fa0707f7..8b145dc3 100644
--- a/backend/decky_loader/wsrouter.py
+++ b/backend/decky_loader/wsrouter.py
@@ -1,4 +1,3 @@
-from _typeshed import DataclassInstance
 from logging import getLogger
 
 from asyncio import AbstractEventLoop, create_task
@@ -27,8 +26,8 @@ class MessageType(IntEnum):
 # WSMessage with slightly better typings
 class WSMessageExtra(WSMessage):
     # TODO message typings here too
-    data: Any
-    type: WSMsgType
+    data: Any # type: ignore yes you can extend it
+    type: WSMsgType # type: ignore
 
 # see wsrouter.ts for typings
 
@@ -133,7 +132,6 @@ class WSRouter:
         self.logger.debug('Firing frontend event %s with args %s', data)
         sent_data: Dict[Any, Any] | None = cast(Dict[Any, Any], data)
         if is_dataclass(data):
-            data_as_dataclass = cast(DataclassInstance, data)
-            sent_data = asdict(data_as_dataclass)
+            sent_data = asdict(data) # type: ignore Argument of type "DataclassInstance | type[DataclassInstance]" cannot be assigned to parameter "obj" of type "DataclassInstance" in function "asdict"
 
         await self.write({ "type": MessageType.EVENT.value, "event": event, "data": sent_data })