Simplify should_dispatch_immediately

ipython · Mar 31, 2021 · ae7637b · ae7637b
1 parent 0ab59fc
commit ae7637b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/ipykernel/ipkernel.py b/ipykernel/ipkernel.py
@@ -556,7 +556,7 @@ def do_clear(self):
         self.shell.reset(False)
         return dict(status='ok')
 
-    def should_dispatch_immediately(self, msg, *args):
+    def should_dispatch_immediately(self, msg):
         try:
             msg_type = msg['header']['msg_type']
             if msg_type in self.comm_msg_types:

diff --git a/ipykernel/kernelbase.py b/ipykernel/kernelbase.py
@@ -399,9 +399,7 @@ def dispatch_queue(self):
     def _message_counter_default(self):
         return itertools.count()
 
-    def should_dispatch_immediately(
-        self, msg, idents, dispatch
-    ):
+    def should_dispatch_immediately(self, msg):
         """
         This provides a hook for dispatching incoming messages
         from the frontend immediately, and out of order.
@@ -423,9 +421,7 @@ def schedule_dispatch(self, msg, dispatch):
 
         new_args = (msg, idents)
 
-        if self.should_dispatch_immediately(
-            msg, idents, dispatch
-        ):
+        if self.should_dispatch_immediately(msg):
             return self.io_loop.add_callback(dispatch, *new_args)
 
         idx = next(self._message_counter)