lint

basetenlabs · Oct 17, 2024 · 2d12f4f · 2d12f4f
1 parent e5ded78
commit 2d12f4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/truss/tests/test_trussless_custom_server.py b/truss/tests/test_trussless_custom_server.py
@@ -23,7 +23,7 @@ def test_custom_server_truss():
                 local_port=8090,
                 detach=True,
                 wait_for_server_ready=True,
-                model_server_stop_retry_override=stop_after_attempt(5),
+                model_server_stop_retry_override=stop_after_attempt(3),
             )
         except Exception as e:
             raise Exception(f"Failed to start container: {e}")

diff --git a/truss/truss_handle.py b/truss/truss_handle.py
@@ -1072,7 +1072,7 @@ def _wait_for_docker_build(container) -> None:
                 raise ContainerIsDownError(f"Container stuck in state: {state.value}.")
 
 
-def _wait_for_model_server(url: str, stop=stop_after_delay(120)) -> Response:
+def _wait_for_model_server(url: str, stop=stop_after_delay(120)) -> Response:  # type: ignore[return]
     for attempt in Retrying(
         stop=stop,
         wait=wait_fixed(2),
@@ -1083,7 +1083,6 @@ def _wait_for_model_server(url: str, stop=stop_after_delay(120)) -> Response:
     ):
         with attempt:
             response = requests.get(url)
-            print(f"DEBUG response status: {response.status_code}")
             return response