Disable failing integration tests

Author: pintaoz-aws

tests/integ/test_tf.py

@@ -85,6 +85,9 @@ def test_mnist_with_checkpoint_config(
     tensorflow_training_latest_version,
     tensorflow_training_latest_py_version,
 ):
+    if Version(tensorflow_training_latest_version) >= Version("2.16"):
+        pytest.skip("This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
+                    "https://github.com/tensorflow/io/issues/2039")
     checkpoint_s3_uri = "s3://{}/checkpoints/tf-{}".format(
         sagemaker_session.default_bucket(), sagemaker_timestamp()
     )
@@ -235,6 +238,9 @@ def test_mnist_distributed_cpu(
     tensorflow_training_latest_version,
     tensorflow_training_latest_py_version,
 ):
+    if Version(tensorflow_training_latest_version) >= Version("2.16"):
+        pytest.skip("This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
+                    "https://github.com/tensorflow/io/issues/2039")
     _create_and_fit_estimator(
         sagemaker_session,
         tensorflow_training_latest_version,

tests/integ/test_tuner.py

@@ -12,6 +12,7 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+from distutils.version import Version
 import json
 import os
 import time
@@ -691,6 +692,9 @@ def test_tuning_tf(
     tensorflow_training_latest_version,
     tensorflow_training_latest_py_version,
 ):
+    if Version(tensorflow_training_latest_version) >= Version("2.16"):
+        pytest.skip("This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
+                    "https://github.com/tensorflow/io/issues/2039")
     resource_path = os.path.join(DATA_DIR, "tensorflow_mnist")
     script_path = "mnist.py"