Skip to content

Commit

Permalink
sleep 3
Browse files Browse the repository at this point in the history
  • Loading branch information
Tianshu Cheng authored and Tianshu Cheng committed Dec 2, 2024
1 parent 657200c commit 805a2a5
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions truss/tests/test_model_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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={})
Expand Down

0 comments on commit 805a2a5

Please sign in to comment.