Skip to content

Commit b22ad06

Browse files
Captainiajiapinw
authored andcommitted
chore: skip flaky test (aws#4535)
* chore: skip flaky test * skip deprecated models test
1 parent 939b59e commit b22ad06

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/unit/sagemaker/jumpstart/test_notebook_utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -518,6 +518,10 @@ def get_manifest_more_versions(region: str = JUMPSTART_DEFAULT_REGION_NAME):
518518
list_old_models=False, list_versions=True
519519
) == list_jumpstart_models(list_versions=True)
520520

521+
@pytest.mark.skipif(
522+
datetime.datetime.now() < datetime.datetime(year=2024, month=5, day=1),
523+
reason="Contact JumpStart team to fix flaky test.",
524+
)
521525
@patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor._get_manifest")
522526
@patch("sagemaker.jumpstart.notebook_utils.DEFAULT_JUMPSTART_SAGEMAKER_SESSION.read_s3_file")
523527
def test_list_jumpstart_models_vulnerable_models(
@@ -570,6 +574,10 @@ def vulnerable_training_model_spec(bucket, key, *args, **kwargs):
570574

571575
assert patched_read_s3_file.call_count == 0
572576

577+
@pytest.mark.skipif(
578+
datetime.datetime.now() < datetime.datetime(year=2024, month=5, day=1),
579+
reason="Contact JumpStart team to fix flaky test.",
580+
)
573581
@patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor._get_manifest")
574582
@patch("sagemaker.jumpstart.notebook_utils.DEFAULT_JUMPSTART_SAGEMAKER_SESSION.read_s3_file")
575583
def test_list_jumpstart_models_deprecated_models(

0 commit comments

Comments
 (0)