diff --git a/spalloc_client/protocol_client.py b/spalloc_client/protocol_client.py index 5706ecb3d..032c3a339 100644 --- a/spalloc_client/protocol_client.py +++ b/spalloc_client/protocol_client.py @@ -19,8 +19,6 @@ import json import socket from threading import current_thread, RLock, local -from spinn_utilities.abstract_context_manager import AbstractContextManager -from spinn_utilities.overrides import overrides from spalloc_client._utils import time_left, timed_out, make_timeout @@ -51,7 +49,7 @@ def __init__(self): self.sock = None -class ProtocolClient(AbstractContextManager): +class ProtocolClient(object): """ A simple (blocking) client implementation of the `spalloc-server `_ protocol. @@ -105,10 +103,6 @@ def __init__(self, hostname, port=22244, timeout=None): self._notifications_lock = RLock() self._default_timeout = timeout - @overrides(AbstractContextManager._context_entered) - def _context_entered(self): # pragma: no cover - self.connect(self._default_timeout) - def _get_connection(self, timeout): if self._dead: raise OSError(errno.ENOTCONN, "not connected")