diff --git a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py index 5d31628a..78ff562c 100644 --- a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py +++ b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py @@ -210,8 +210,14 @@ def test_multiple_inputs(self): def test_any_composite(self): # TODO: b/294854329 - Re-enable when the TF fuzzing issue is resolved. - if tf.__version__.startswith('2.15') or tf.__version__.startswith('2.16'): - self.skipTest('Fails with tf internal error on tf 2.15, 2.16 nightly') + if ( + tf.__version__.startswith('2.15') + or tf.__version__.startswith('2.16') + or tf.__version__.startswith('2.17') + ): + self.skipTest( + 'Fails with tf internal error on tf 2.15, 2.16, 2.17 nightly' + ) i = tf.keras.Input([2], name='input') o = i