Skip to content

Commit ee2e507

Browse files
Captainiaroot
authored and
root
committed
fix: skip flakey tests pending investigation (aws#4667)
1 parent f03af5e commit ee2e507

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/unit/sagemaker/jumpstart/test_notebook_utils.py

+5
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from __future__ import absolute_import
22
import json
3+
import datetime
34

45
from unittest import TestCase
56
from unittest.mock import Mock, patch
@@ -514,6 +515,10 @@ def get_manifest_more_versions(region: str = JUMPSTART_DEFAULT_REGION_NAME):
514515
list_old_models=False, list_versions=True
515516
) == list_jumpstart_models(list_versions=True)
516517

518+
@pytest.mark.skipif(
519+
datetime.datetime.now() < datetime.datetime(year=2024, month=8, day=1),
520+
reason="Contact JumpStart team to fix flaky test.",
521+
)
517522
@patch("sagemaker.jumpstart.accessors.JumpStartModelsAccessor._get_manifest")
518523
@patch("sagemaker.jumpstart.notebook_utils.DEFAULT_JUMPSTART_SAGEMAKER_SESSION.read_s3_file")
519524
def test_list_jumpstart_models_vulnerable_models(

0 commit comments

Comments
 (0)