def METHOD_NAME(
    self, strategy, use_dataset_creator
):
    model = self._model_fit(
        strategy, use_dataset_creator=use_dataset_creator
    )
    strategy = model.distribute_strategy
    self.assertIs(
        strategy._cluster_coordinator,
        tf.distribute.experimental.coordinator.ClusterCoordinator(strategy),
    )