Skip to content

Commit baad38e

Browse files
author
pialidas
committed
fix the test
1 parent 68b38d0 commit baad38e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integ/test_auto_ml.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,8 @@
3838

3939
# use a succeeded AutoML job to test describe and list candidates method, otherwise tests will run too long
4040
AUTO_ML_JOB_NAME = "python-sdk-integ-test-base-job"
41+
DEFAULT_MODEL_NAME = "python-sdk-automl"
42+
4143

4244
EXPECTED_DEFAULT_JOB_CONFIG = {
4345
"CompletionCriteria": {"MaxCandidates": 3},
@@ -287,8 +289,9 @@ def test_create_model_best_candidate(sagemaker_session, cpu_instance_type):
287289
auto_ml = AutoML.attach(job_name=AUTO_ML_JOB_NAME, sagemaker_session=sagemaker_session)
288290
best_candidate = auto_ml.best_candidate()
289291

290-
with timeout(minutes=2):
292+
with timeout(minutes=5):
291293
pipeline_model = auto_ml.create_model(
294+
name=DEFAULT_MODEL_NAME,
292295
candidate=best_candidate,
293296
sagemaker_session=sagemaker_session,
294297
vpc_config=None,

0 commit comments

Comments
 (0)