Disabling flaky test within //third_party/py/keras/integration_test:gradient_checkpoint_test_xla_gpu.
Since only `test_raises_oom_exception` was flaky, and it was not a special testing configuration (asan/tsan), I skipped that one test case via code instead of the entire test via BUILD file.
@@ -143,6 +143,7 @@ def _train_with_recompute(n_steps):
class GradientCheckpointTest(tf.test.TestCase):
def test_raises_oom_exception(self):
+ self.skipTest('b/232015009: flaky test')
if not _limit_gpu_memory():
self.skipTest('No virtual GPUs found')
with self.assertRaises(Exception) as context:
