diff --git a/tensorflow/python/distribute/ctl_correctness_test.py b/tensorflow/python/distribute/ctl_correctness_test.py index 59fae808c21..6ce6a59df47 100644 --- a/tensorflow/python/distribute/ctl_correctness_test.py +++ b/tensorflow/python/distribute/ctl_correctness_test.py @@ -224,7 +224,7 @@ class TestDistributionStrategyDnnCorrectness(test.TestCase, @combinations.generate( combinations.combine( distribution=strategy_combinations.all_strategies, - optimizer_fn=strategy_combinations.optimizers_v1_and_v2, + optimizer_fn=strategy_combinations.optimizers_v2, mode=['eager'], iteration_type=['iterator', 'dataset'], inside_func=[False, True],