Merge pull request #117 from Becksteinlab/jupyter-error

Accept arbitrary args in signal handler

Author: ljwoods2

CHANGELOG.md

@@ -24,6 +24,8 @@ The rules for this file:
 ### Added
 
 ### Fixed
+* Changed signal handler method to accept arbitrary args to prevent
+  error being raised in new jupyter version (Issue #116, PR #117)
 * Removed timeout changes during pause/resume for IMDv2 and IMv3
   (Issue #96, PR #118)
 

imdclient/IMDClient.py

@@ -73,7 +73,6 @@ def __init__(
         continue_after_disconnect=None,
         **kwargs,
     ):
-
         self._stopped = False
         self._conn = self._connect_to_server(host, port, socket_bufsize)
         self._imdsinfo = self._await_IMD_handshake()
@@ -147,7 +146,7 @@ def __init__(
 
             self._producer.start()
 
-    def signal_handler(self):
+    def signal_handler(self, *args, **kwargs):
         """Catch SIGINT to allow clean shutdown on CTRL+C.
 
         This also ensures that main thread execution doesn't get stuck
@@ -484,7 +483,6 @@ def run(self):
             self._buf.notify_producer_finished()
 
     def _expect_header(self, expected_type, expected_value=None):
-
         header = self._get_header()
 
         if header.type != expected_type:
@@ -934,7 +932,6 @@ def notify_consumer_finished(self):
 
 
 class IMDFrame:
-
     def __init__(self, n_atoms, imdsinfo):
         if imdsinfo.time:
             self.time = 0.0