Skip to content

Commit 8f736c9

Browse files
author
Basil Beirouti
committed
fixed model naming to use unique_name_from_base
1 parent 4d9d7b8 commit 8f736c9

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/integ/test_marketplace.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -234,11 +234,11 @@ def iris_image(sagemaker_session):
234234

235235

236236
def test_create_model_package(sagemaker_session, boto_session, iris_image):
237-
237+
MODEL_NAME = "iris-classifier-mp"
238238
# Prepare
239239
s3_bucket = sagemaker_session.default_bucket()
240240

241-
model_name = "my-flower-detection-model"
241+
model_name = unique_name_from_base(MODEL_NAME)
242242
model_description = "This model accepts petal length, petal width, sepal length, sepal width and predicts whether \
243243
flower is of type setosa, versicolor, or virginica"
244244

@@ -311,10 +311,10 @@ def test_create_model_package(sagemaker_session, boto_session, iris_image):
311311
# wait for model execution to complete
312312
time.sleep(60 * 3)
313313

314-
# query for all model packages with the name "my-flower-detection-model"
314+
# query for all model packages with the name <MODEL_NAME>
315315
response = sm_client.list_model_packages(
316316
MaxResults=10,
317-
NameContains="my-flower-detection-model",
317+
NameContains=MODEL_NAME,
318318
SortBy="CreationTime",
319319
SortOrder="Descending",
320320
)
@@ -325,7 +325,6 @@ def test_create_model_package(sagemaker_session, boto_session, iris_image):
325325
# assert that response is non-empty
326326
assert len(response["ModelPackageSummaryList"]) > 0
327327

328-
329328
@pytest.mark.skipif(
330329
tests.integ.test_region() in tests.integ.NO_MARKET_PLACE_REGIONS,
331330
reason="Marketplace is not available in {}".format(tests.integ.test_region()),

0 commit comments

Comments
 (0)