diff --git a/piker/data/_web_bs.py b/piker/data/_web_bs.py index da4f2a30..c42b7afa 100644 --- a/piker/data/_web_bs.py +++ b/piker/data/_web_bs.py @@ -26,6 +26,7 @@ import json import trio import trio_websocket +from wsproto.utilities import LocalProtocolError from trio_websocket._impl import ( ConnectionClosed, DisconnectionTimeout, @@ -51,6 +52,7 @@ class NoBsWs: ConnectionRejected, HandshakeError, ConnectionTimeout, + LocalProtocolError, ) def __init__(