diff --git a/truss/tests/test_model_inference.py b/truss/tests/test_model_inference.py index 907f6b9a8..f41562de3 100644 --- a/truss/tests/test_model_inference.py +++ b/truss/tests/test_model_inference.py @@ -245,7 +245,7 @@ def test_requirements_file_truss(test_data_path): truss_dir = test_data_path / "test_requirements_file_truss" tr = TrussHandle(truss_dir) _ = tr.docker_run(local_port=8090, detach=True, wait_for_server_ready=True) - time.sleep(3) + time.sleep(3) # Sleeping to allow the load to finish # The prediction imports torch which is specified in a requirements.txt and returns if GPU is available. response = requests.post(PREDICT_URL, json={})