File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff 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)
7475def 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)
114114def test_huggingface_inference (
115115 sagemaker_session ,
You can’t perform that action at this time.
0 commit comments