Skip to content

Commit 14b2af7

Browse files
fix: tag more integs as flaky for auto-retry (#3912)
1 parent 62326f2 commit 14b2af7

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

tests/integ/test_marketplace.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,6 @@ def test_marketplace_estimator(sagemaker_session, cpu_instance_type):
108108
print(predictor.predict(test_x.values).decode("utf-8"))
109109

110110

111-
@pytest.mark.skip(reason="This test is currently failing. Skipping until fixed")
112111
@pytest.mark.skipif(
113112
tests.integ.test_region() in tests.integ.NO_MARKET_PLACE_REGIONS,
114113
reason="Marketplace is not available in {}".format(tests.integ.test_region()),
@@ -166,6 +165,7 @@ def test_marketplace_attach(sagemaker_session, cpu_instance_type):
166165
tests.integ.test_region() in tests.integ.NO_MARKET_PLACE_REGIONS,
167166
reason="Marketplace is not available in {}".format(tests.integ.test_region()),
168167
)
168+
@pytest.mark.flaky(reruns=5, reruns_delay=2)
169169
def test_marketplace_model(sagemaker_session, cpu_instance_type):
170170
region = sagemaker_session.boto_region_name
171171
account = REGION_ACCOUNT_MAP[region]
@@ -328,7 +328,6 @@ def test_create_model_package(sagemaker_session, boto_session, iris_image):
328328
assert len(response["ModelPackageSummaryList"]) > 0
329329

330330

331-
@pytest.mark.skip(reason="This test is currently failing. Skipping until fixed")
332331
@pytest.mark.skipif(
333332
tests.integ.test_region() in tests.integ.NO_MARKET_PLACE_REGIONS,
334333
reason="Marketplace is not available in {}".format(tests.integ.test_region()),
@@ -372,7 +371,6 @@ def test_marketplace_tuning_job(sagemaker_session, cpu_instance_type):
372371
tuner.wait()
373372

374373

375-
@pytest.mark.skip(reason="This test is currently failing. Skipping until fixed")
376374
@pytest.mark.skipif(
377375
tests.integ.test_region() in tests.integ.NO_MARKET_PLACE_REGIONS,
378376
reason="Marketplace is not available in {}".format(tests.integ.test_region()),

tests/integ/test_model_card.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ def binary_classifier_fixture(
125125
assert "Could not find model" in str(exception.value)
126126

127127

128+
@pytest.mark.flaky(reruns=5, reruns_delay=2)
128129
def test_model_card_create_read_update_and_delete(
129130
sagemaker_session,
130131
binary_classifier,

0 commit comments

Comments
 (0)