Added ServerNotReadyError to the list of errors caught by wait_for_server_initialization()

Author: apockill

brainframe/api/stub.py

@@ -3,7 +3,7 @@
 
 from requests.exceptions import ConnectionError, ReadTimeout
 
-from .bf_errors import UnauthorizedError, UnknownError
+from .bf_errors import UnauthorizedError, UnknownError, ServerNotReadyError
 from . import stubs
 from .stubs.base_stub import DEFAULT_TIMEOUT
 
@@ -59,8 +59,8 @@ def wait_for_server_initialization(self, timeout: float = None):
             try:
                 # Test connection to server
                 self.version()
-            except (ConnectionError, ConnectionRefusedError,
-                    UnauthorizedError, ReadTimeout):
+            except (ConnectionError, ConnectionRefusedError, ReadTimeout,
+                    UnauthorizedError, ServerNotReadyError):
                 # Server not started yet or there is a communication
                 # error
                 pass