From e235c2f3a08b621d6eac9647488d375fa0212105 Mon Sep 17 00:00:00 2001
From: pintaoz-aws <pintaozou.sde@gmail.com>
Date: Fri, 30 Aug 2024 09:52:51 -0700
Subject: [PATCH] Disable test_mnist_async

---
 tests/integ/test_tf.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tests/integ/test_tf.py b/tests/integ/test_tf.py
index 8befd0efeb..db44acd5dc 100644
--- a/tests/integ/test_tf.py
+++ b/tests/integ/test_tf.py
@@ -306,6 +306,11 @@ def _create_and_fit_estimator(sagemaker_session, tf_version, py_version, instanc
 
 @pytest.mark.slow_test
 def test_mnist_async(sagemaker_session, cpu_instance_type, tf_full_version, tf_full_py_version):
+    if Version(tf_full_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"
+        )
     if tf_full_version == "2.7.0":
         tf_full_version = "2.7"