[TEST] TF 2.16 sm training skip s3 plugin test due to tensorflow-io (#…

…4200) * [TEST] TF 2.16 sm training toolkit upgrade * skip s3_plugin test * revert toml * formatting
aws · Sep 5, 2024 · 97ad513 · 97ad513
1 parent bed46e8
commit 97ad513
Showing 1 changed file with 1 addition and 0 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,6 +268,7 @@ 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")