Disable test_mnist_async (#4858)

aws · Aug 30, 2024 · 4ab6c84 · 4ab6c84
1 parent 367245b
commit 4ab6c84
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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"