Update loss_scale_optimizer_test.py
@@ -1167,7 +1167,7 @@ class LossScaleOptimizerTest(tf.test.TestCase, parameterized.TestCase):
self.assertEqual(self._eval_if_tensor(opt.inner_optimizer.momentum), 0.5)
self.assertEqual(self.evaluate(opt.loss_scale), 2.)
self.assertEqual(opt.dynamic_growth_steps, 3.)
- self.assertEqual(opt.dynamic, 4.)
+ self.assertTrue(opt.dynamic)
if lso_type in ('v1', 'v2'):
# Deserializing a LossScaleOptimizerV1 results in a V2 LossScaleOptimizer.
self.assertEqual(type(opt), loss_scale_optimizer.LossScaleOptimizer)
