Skip to content

Commit 36350d1

Browse files
committed
fix: integ tests
1 parent 61e0061 commit 36350d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_async_inference.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -100,11 +100,11 @@ def test_async_walkthrough(sagemaker_session, cpu_instance_type, training_set):
100100
result_not_wait = predictor_async.predict_async(input_path=input_s3_path)
101101
assert isinstance(result_not_wait, AsyncInferenceResponse)
102102
assert result_not_wait.output_path.startswith("s3://" + sagemaker_session.default_bucket())
103-
time.sleep(5)
104-
result_not_wait = result_not_wait.get_result()
105103
assert result_not_wait.failure_path.startswith(
106104
"s3://" + sagemaker_session.default_bucket() + "/async-endpoint-failures/"
107105
)
106+
time.sleep(5)
107+
result_not_wait = result_not_wait.get_result()
108108
assert len(result_not_wait) == 5
109109
for record in result_not_wait:
110110
assert record.label["projection"] is not None

0 commit comments

Comments
 (0)