Skip to content

Commit 8a42456

Browse files
fix: make marketplace jobnames random
1 parent 1382f15 commit 8a42456

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

tests/integ/marketplace_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
# ANY KIND, either express or implied. See the License for the specific
1212
# language governing permissions and limitations under the License.
1313
from __future__ import absolute_import
14+
import uuid
1415

1516
REGION_ACCOUNT_MAP = {
1617
"us-east-1": "865070037744",
@@ -34,3 +35,7 @@
3435
"cn-north-1": "295401494951",
3536
"cn-northwest-1": "304690803264",
3637
}
38+
39+
40+
def random_8_digit_alpha_numeric_gen():
41+
return str(uuid.uuid4().hex.upper()[0:8])

tests/integ/test_marketplace.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from sagemaker.utils import _aws_partition
2929
from tests.integ import DATA_DIR
3030
from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
31-
from tests.integ.marketplace_utils import REGION_ACCOUNT_MAP
31+
from tests.integ.marketplace_utils import REGION_ACCOUNT_MAP, random_8_digit_alpha_numeric_gen
3232

3333

3434
# All these tests require a manual 1 time subscription to the following Marketplace items:
@@ -117,7 +117,7 @@ def test_marketplace_attach(sagemaker_session, cpu_instance_type):
117117
instance_count=1,
118118
instance_type=cpu_instance_type,
119119
sagemaker_session=sagemaker_session,
120-
base_job_name="test-marketplace",
120+
base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
121121
)
122122

123123
train_input = mktplace.sagemaker_session.upload_data(
@@ -205,7 +205,7 @@ def test_marketplace_tuning_job(sagemaker_session, cpu_instance_type):
205205
instance_count=1,
206206
instance_type=cpu_instance_type,
207207
sagemaker_session=sagemaker_session,
208-
base_job_name="test-marketplace",
208+
base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
209209
)
210210

211211
train_input = mktplace.sagemaker_session.upload_data(
@@ -218,7 +218,7 @@ def test_marketplace_tuning_job(sagemaker_session, cpu_instance_type):
218218

219219
tuner = HyperparameterTuner(
220220
estimator=mktplace,
221-
base_tuning_job_name="byo",
221+
base_tuning_job_name="byo" + random_8_digit_alpha_numeric_gen(),
222222
objective_metric_name="validation:accuracy",
223223
hyperparameter_ranges=hyperparameter_ranges,
224224
max_jobs=2,
@@ -248,7 +248,7 @@ def test_marketplace_transform_job(sagemaker_session, cpu_instance_type):
248248
instance_count=1,
249249
instance_type=cpu_instance_type,
250250
sagemaker_session=sagemaker_session,
251-
base_job_name="test-marketplace",
251+
base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
252252
)
253253

254254
train_input = algo.sagemaker_session.upload_data(

0 commit comments

Comments
 (0)