1 files changed, 17 insertions, 17 deletions
diff --git a/kennel/main.py b/kennel/main.py
index fc0fd0e..c1106fe 100644
--- a/kennel/main.py
+++ b/kennel/main.py
@@ -1,37 +1,37 @@
+import asyncio
+import uuid
+from typing import Annotated, Optional
+
 from fastapi import (
-    FastAPI,
+    Cookie,
+    Depends,
     Request,
     Response,
     WebSocket,
+    WebSocketDisconnect,
     WebSocketException,
     status,
-    Cookie,
-    Depends,
 )
 from fastapi.responses import FileResponse
 from fastapi.staticfiles import StaticFiles
-from kennel.engine.systems.system import SystemType
+
+from kennel.app import app, logger
+from kennel.config import config
 from kennel.engine.systems.network import (
+    EntityBornEvent,
     Event,
-    Publishable,
     EventType,
     InitialStateEvent,
+    Publishable,
     SetControllableEvent,
-    EntityBornEvent,
 )
-from typing import Annotated, Optional
+from kennel.engine.systems.system import SystemType
 from kennel.kennel import (
+    create_session_controllable_entities,
+    entity_manager,
     kennel,
     system_manager,
-    entity_manager,
-    create_session_controllable_entities,
 )
-from kennel.app import app, logger
-from kennel.kennelcats import KennelCatService
-from kennel.middleware import logger_middleware
-from kennel.config import config
-import asyncio
-import uuid
 
 app.mount("/static", StaticFiles(directory="static/dist"), name="static")
 
@@ -118,7 +118,7 @@ async def websocket_endpoint(
         network_system.add_client(session, WebSocketClient(websocket))
         while True:
             message = await websocket.receive_json()
-            if type(message) is not list:
+            if not isinstance(message, list):
                 message = [message]
             events = [Event.from_dict(event) for event in message]
             if not all([event is not None for event in events]):
@@ -126,7 +126,7 @@ async def websocket_endpoint(
                 continue
             for event in events:
                 network_system.client_event(session, event)
-    except WebSocketDisconnect as e:
+    except WebSocketDisconnect:
         logger.info(f"Websocket connection closed by client: {session}")
     except Exception as e:
         logger.error("Exception occurred", exc_info=e)