Skip to content

Commit 506083a

Browse files
committed
skipping hf inference test
1 parent f44062a commit 506083a

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

tests/integ/test_huggingface.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,9 @@ def test_huggingface_training(
6868
hf.fit(train_input)
6969

7070

71-
@pytest.mark.skip(
72-
reason="need to re enable it later",
71+
@pytest.mark.release
72+
@pytest.mark.skipif(
73+
integ.test_region() in integ.TRAINING_NO_P2_REGIONS, reason="no ml.p2 instances in this region"
7374
)
7475
def test_huggingface_training_tf(
7576
sagemaker_session,
@@ -106,10 +107,8 @@ def test_huggingface_training_tf(
106107

107108
hf.fit(train_input)
108109

109-
110-
@pytest.mark.release
111-
@pytest.mark.skipif(
112-
integ.test_region() in integ.TRAINING_NO_P2_REGIONS, reason="no ml.p2 instances in this region"
110+
@pytest.mark.skip(
111+
reason="need to re enable it later",
113112
)
114113
def test_huggingface_inference(
115114
sagemaker_session,

0 commit comments

Comments
 (0)