diff --git a/tensorflow/python/keras/distribute/custom_training_loop_models_test.py b/tensorflow/python/keras/distribute/custom_training_loop_models_test.py index b6b92391cef..fe80a56aa75 100644 --- a/tensorflow/python/keras/distribute/custom_training_loop_models_test.py +++ b/tensorflow/python/keras/distribute/custom_training_loop_models_test.py @@ -204,11 +204,6 @@ class KerasModelsTest(test.TestCase, parameterized.TestCase): train_step(input_iterator) # TODO(b/165912857): Re-enable. - @combinations.generate( - combinations.combine( - distribution=strategy_combinations.all_strategies, - mode=["eager"] - )) def DISABLED_test_lstm(self, distribution): batch_size = 32