diff --git a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py index cd171703..5d31628a 100644 --- a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py +++ b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py @@ -210,8 +210,8 @@ 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'): - self.skipTest('Fails with tf internal error on tf 2.14 nightly') + 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') i = tf.keras.Input([2], name='input') o = i