Merge pull request #273 from BQSKit/change_server_close_error

Handle EOFError and ConnectionResetError without chaining exceptions in server connection failure
BQSKit · Sep 8, 2024 · 5176210 · 5176210
2 parents e05f947 + 9fc4126
commit 5176210
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bqskit/compiler/compiler.py b/bqskit/compiler/compiler.py
@@ -415,7 +415,12 @@ def _send(self, msg: RuntimeMessage, payload: Any) -> None:
         except Exception as e:
             self.conn = None
             self.close()
-            raise RuntimeError('Server connection unexpectedly closed.') from e
+            if isinstance(e, (EOFError, ConnectionResetError)):
+                raise RuntimeError('Server connection unexpectedly closed.')
+            else:
+                raise RuntimeError(
+                    'Server connection unexpectedly closed.',
+                ) from e
 
     def _send_recv(
         self,