re-enable the test since the issue is fixed

PiperOrigin-RevId: 330594702
Change-Id: Ic510e208cee5302adafa823f0fa3e539de835ae1
This commit is contained in:
Yanhua Sun 2020-09-08 15:03:40 -07:00 committed by TensorFlower Gardener
parent 60bee51060
commit 4d343e5441

View File

@ -203,8 +203,7 @@ class KerasModelsTest(test.TestCase, parameterized.TestCase):
train_step(input_iterator)
# TODO(b/165912857): Re-enable.
def DISABLED_test_lstm(self, distribution):
def test_lstm(self, distribution):
batch_size = 32