We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f44062a commit 1db6069Copy full SHA for 1db6069
tests/integ/test_huggingface.py
@@ -68,8 +68,9 @@ def test_huggingface_training(
68
hf.fit(train_input)
69
70
71
-@pytest.mark.skip(
72
- reason="need to re enable it later",
+@pytest.mark.release
+@pytest.mark.skipif(
73
+ integ.test_region() in integ.TRAINING_NO_P2_REGIONS, reason="no ml.p2 instances in this region"
74
)
75
def test_huggingface_training_tf(
76
sagemaker_session,
@@ -107,9 +108,8 @@ def test_huggingface_training_tf(
107
108
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"
+@pytest.mark.skip(
+ reason="need to re enable it later",
113
114
def test_huggingface_inference(
115
0 commit comments