From 805a2a51a79068c5eca90798697b3fe0342f215d Mon Sep 17 00:00:00 2001 From: Tianshu Cheng Date: Mon, 2 Dec 2024 10:28:41 -0800 Subject: [PATCH] sleep 3 --- truss/tests/test_model_inference.py | 1 + 1 file changed, 1 insertion(+) diff --git a/truss/tests/test_model_inference.py b/truss/tests/test_model_inference.py index 6bfcc005a..907f6b9a8 100644 --- a/truss/tests/test_model_inference.py +++ b/truss/tests/test_model_inference.py @@ -245,6 +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) # The prediction imports torch which is specified in a requirements.txt and returns if GPU is available. response = requests.post(PREDICT_URL, json={})