message: Add MWMS combinations to custom_training_loop_metrics_test.
diff: @@ -34,7 +34,8 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
  @combinations.generate(
  combinations.combine(
  - distribution=strategy_combinations.all_strategies,
  + distribution=strategy_combinations.all_strategies +
  + strategy_combinations.multiworker_strategies,
  mode=["eager"]
  ))
  def test_multiple_keras_metrics_experimental_run(self, distribution):
  @@ -58,7 +59,8 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
  @combinations.generate(
  combinations.combine(
  - distribution=strategy_combinations.all_strategies,
  + distribution=strategy_combinations.all_strategies+
  + strategy_combinations.multiworker_strategies,
  mode=["eager"]
  ))
  def test_update_keras_metric_declared_in_strategy_scope(self, distribution):
  @@ -98,4 +100,4 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
  if __name__ == "__main__":
  - test.main()
  + combinations.main()
  