Fix fixture addition from PR aws#4200

arjkesh · Oct 22, 2024 · 5ca4a50 · 5ca4a50
1 parent e5de0c2
commit 5ca4a50
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_mnist.py b/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_mnist.py
@@ -268,12 +268,10 @@ def _test_distributed_mnist_custom_ps(
     _assert_checkpoint_exists_v2(sagemaker_session.boto_region_name, estimator.model_dir, 10)
 
 
-@pytest.mark.skip_tf216_only
-@pytest.mark.skipif(is_pr_context(), reason=SKIP_PR_REASON)
 @pytest.mark.model("mnist")
 @pytest.mark.integration("s3 plugin")
 @pytest.mark.team("frameworks")
-def test_s3_plugin(ecr_image, sagemaker_regions, instance_type, framework_version):
+def test_s3_plugin(ecr_image, sagemaker_regions, instance_type, framework_version, skip_tf216_only):
     invoke_sm_helper_function(
         ecr_image, sagemaker_regions, _test_s3_plugin_function, instance_type, framework_version
     )