fix linter issue and code format

Signed-off-by: s-gatti <[email protected]>
Signed-off-by: Charles Kim <[email protected]>

Author: s-gatti

app/socket_connection_manager.py

@@ -95,7 +95,8 @@ async def broadcast(self, message: Union[str, dict, list]) -> None:
                         await websocket.close()
                     logger.warning(
                         f'Failed to send message: "{message}"'
-                        f' to websocket: "{websocket}", connection closed."')
+                        f' to websocket: "{websocket}", connection closed."'
+                    )
                 except RuntimeError as e:
                     logger.warning(
                         f'Failed to send: "{message}" to websocket: "{websocket}."',
@@ -112,7 +113,7 @@ async def received_message(self, websocket: WebSocket, message: str) -> None:
                 websocket=websocket, message=INVALID_JSON_ERROR_STR
             )
 
-    async def relay_video_frames(self, connection: WebSocketConnection):
+    async def relay_video_frames(self, connection: WebSocketConnection) -> None:
         if connection.type == WebSocketTypeEnum.VIDEO:
             websocket = connection.websocket
             try:
@@ -131,10 +132,10 @@ async def relay_video_frames(self, connection: WebSocketConnection):
                 logger.error(f"Failed with {e}")
             finally:
                 await websocket.close()
-                self.active_connections.remove(connection)
+                self.disconnect(connection)
         else:
             logger.error(
-                f'Excpected websocket connection of type {WebSocketTypeEnum.VIDEO}'
+                f"Expected websocket connection of type {WebSocketTypeEnum.VIDEO}"
             )
 
     # Note: Currently we only support one message handler per type, registering the