From 7c5b1779f8ecb79c7da8049094fb3f40b0b064f1 Mon Sep 17 00:00:00 2001
From: Akshay Modi <nareshmodi@google.com>
Date: Wed, 3 Jul 2019 14:34:31 -0700
Subject: [PATCH] Actually disable the flaky test

PiperOrigin-RevId: 256439799
---
 tensorflow/python/keras/distribute/BUILD | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD
index ab919ca68a6..d887bd589ad 100644
--- a/tensorflow/python/keras/distribute/BUILD
+++ b/tensorflow/python/keras/distribute/BUILD
@@ -137,9 +137,6 @@ py_library(
         "keras_lstm_model_correctness_test.py",
         "keras_stateful_lstm_model_correctness_test.py",
     ],
-    tags = [
-        "no_oss",
-    ],
     deps = [
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:training",
@@ -223,6 +220,7 @@ distribute_py_test(
     shard_count = 31,
     tags = [
         "multi_and_single_gpu",
+        "no_oss",  # b/136660639
         "no_windows_gpu",
         "notap",  # b/131937016
         "notsan",