message: Fix None gradient test
diff: @@ -253,8 +253,7 @@ class DynamicLossScaleTest(test.TestCase, parameterized.TestCase):
  self._test_helper(inputs, expected_outputs, init_loss_scale)
  @test_util.run_in_graph_and_eager_modes
  - def test_update_with_none_gradients(self, strategy_fn):
  - with strategy_fn().scope():
  + def test_update_with_none_gradients(self):
  loss_scaler = loss_scale_module.DynamicLossScale()
  loss_scaler.update([None])
  