Fix fixture addition from PR #4200 (#4364)

* Fix fixture addition from PR #4200

* Update ['dlc_developer_config.toml']

dlc_developer_config.toml:
{   'build': {   'build_frameworks': ['tensorflow'],
                 'build_inference': False,
                 'build_training': True},
    'buildspec_override': {   'dlc-pr-tensorflow-2-training': 'tensorflow/training/buildspec-2-16-sm.yml'},
    'dev': {   'deep_canary_mode': False,
               'graviton_mode': False,
               'neuronx_mode': False},
    'test': {   'ec2_tests': False,
                'ecs_tests': False,
                'eks_tests': False,
                'sagemaker_local_tests': False,
                'sagemaker_remote_tests': True,
                'sanity_tests': False}}

* add build tag override

* update

* Update buildspec-2-16-sm.yml

* Restore ['dlc_developer_config.toml']

dlc_developer_config.toml:
('Restore to '
 'https://raw.githubusercontent.com/aws/deep-learning-containers/master/dlc_developer_config.toml')

* Restore ['dlc_developer_config.toml']

dlc_developer_config.toml:
('Restore to '
 'https://raw.githubusercontent.com/aws/deep-learning-containers/master/dlc_developer_config.toml')

Author: arjkesh

tensorflow/training/buildspec-2-16-sm.yml

@@ -38,6 +38,7 @@ images:
     tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
     latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION, "-sagemaker" ]
     docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /Dockerfile., *DEVICE_TYPE ]
+    # build_tag_override: "pr:2.16.2-cpu-py310-ubuntu20.04-sagemaker-pr-4362-autopatch"
     target: sagemaker
     enable_test_promotion: true
     context:
@@ -57,6 +58,7 @@ images:
                                 *OS_VERSION, "-sagemaker" ]
     docker_file: !join [ docker/, *SHORT_VERSION, /, *DOCKER_PYTHON_VERSION, /, *CUDA_VERSION,
                          /Dockerfile., *DEVICE_TYPE ]
+    # build_tag_override: "pr:2.16.2-gpu-py310-cu123-ubuntu20.04-sagemaker-pr-4362-autopatch"
     target: sagemaker
     enable_test_promotion: true
     context:

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
     )