diff --git a/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smtrcomp.py b/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smtrcomp.py index 5cb758d72cb4..e339a4adbb39 100644 --- a/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smtrcomp.py +++ b/test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smtrcomp.py @@ -382,6 +382,7 @@ def test_training( captured = capsys.readouterr() _assert_training_compiler_invoked(captured) + @pytest.mark.skip("Skipping this test on TF DLCs") @pytest.mark.model("distilbert") def test_BYOC_training( self,