Two tiny corrections found by mypy 1.13 (#1032)

* Fixed reuse of local route_type as different types * added dict type to self.methods
sparckles · Nov 16, 2024 · 51bf8a3 · 51bf8a3
1 parent 31283b2
commit 51bf8a3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/robyn/processpool.py b/robyn/processpool.py
@@ -188,8 +188,8 @@ def spawn_process(
     for middleware_type, middleware_function in global_middlewares:
         server.add_global_middleware(middleware_type, middleware_function)
 
-    for route_type, endpoint, function in route_middlewares:
-        server.add_middleware_route(route_type, endpoint, function)
+    for http_route_type, endpoint, function in route_middlewares:
+        server.add_middleware_route(http_route_type, endpoint, function)
 
     if Events.STARTUP in event_handlers:
         server.add_startup_handler(event_handlers[Events.STARTUP])

diff --git a/robyn/ws.py b/robyn/ws.py
@@ -23,7 +23,7 @@ class WebSocket:
     def __init__(self, robyn_object: "Robyn", endpoint: str, config: Config = Config(), dependencies: DependencyMap = DependencyMap()) -> None:
         self.robyn_object = robyn_object
         self.endpoint = endpoint
-        self.methods = {}
+        self.methods: dict = {}
         self.config = config
         self.dependencies = dependencies