Skip to content

Commit 1db6069

Browse files
authored
fix: skip hf inference test (#2514)
1 parent f44062a commit 1db6069

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/integ/test_huggingface.py

+5-5
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,
@@ -107,9 +108,8 @@ def test_huggingface_training_tf(
107108
hf.fit(train_input)
108109

109110

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"
111+
@pytest.mark.skip(
112+
reason="need to re enable it later",
113113
)
114114
def test_huggingface_inference(
115115
sagemaker_session,

0 commit comments

Comments
 (0)