Skip TFModelStageTest.test_any_composite for any tf-nightly.

PiperOrigin-RevId: 645345217
tensorflow · Jun 21, 2024 · 00f9191 · 00f9191
1 parent 150da27
commit 00f9191
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py
@@ -214,10 +214,9 @@ def test_any_composite(self):
         tf.__version__.startswith('2.15')
         or tf.__version__.startswith('2.16')
         or tf.__version__.startswith('2.17')
+        or '-dev' in tf.__version__  # tf-nightly
     ):
-      self.skipTest(
-          'Fails with tf internal error on tf 2.15, 2.16, 2.17 nightly'
-      )
+      self.skipTest('Fails with tf internal error since tf 2.15')
 
     i = tf.keras.Input([2], name='input')
     o = i