Skip to content

Commit 760a25b

Browse files
authored
infra: generate test job name at test start instead of module start (#1345)
1 parent b8d5470 commit 760a25b

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/integ/test_auto_ml.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from __future__ import absolute_import
1414

1515
import os
16-
import time
1716

1817
import pytest
1918
import tests.integ
@@ -34,7 +33,7 @@
3433
TRAINING_DATA = os.path.join(DATA_DIR, "iris_training.csv")
3534
TEST_DATA = os.path.join(DATA_DIR, "iris_test.csv")
3635
PROBLEM_TYPE = "MultiClassClassification"
37-
JOB_NAME = "auto-ml-{}".format(time.strftime("%y%m%d-%H%M%S"))
36+
BASE_JOB_NAME = "auto-ml"
3837

3938
# use a succeeded AutoML job to test describe and list candidates method, otherwise tests will run too long
4039
AUTO_ML_JOB_NAME = "python-sdk-integ-test-base-job"
@@ -119,11 +118,11 @@ def test_auto_ml_fit_optional_args(sagemaker_session):
119118
)
120119
inputs = TRAINING_DATA
121120
with timeout(minutes=AUTO_ML_DEFAULT_TIMEMOUT_MINUTES):
122-
auto_ml.fit(inputs, job_name=JOB_NAME)
121+
auto_ml.fit(inputs, job_name=unique_name_from_base(BASE_JOB_NAME))
123122

124-
auto_ml_desc = auto_ml.describe_auto_ml_job(job_name=JOB_NAME)
123+
auto_ml_desc = auto_ml.describe_auto_ml_job(job_name=auto_ml.latest_auto_ml_job.job_name)
125124
assert auto_ml_desc["AutoMLJobStatus"] == "Completed"
126-
assert auto_ml_desc["AutoMLJobName"] == JOB_NAME
125+
assert auto_ml_desc["AutoMLJobName"] == auto_ml.latest_auto_ml_job.job_name
127126
assert auto_ml_desc["AutoMLJobObjective"] == job_objective
128127
assert auto_ml_desc["ProblemType"] == problem_type
129128
assert auto_ml_desc["OutputDataConfig"]["S3OutputPath"] == output_path

0 commit comments

Comments
 (0)