def METHOD_NAME(self, strategy, use_dataset_creator):
    log_dir = self.get_temp_dir()
    callbacks = [callbacks_lib.TensorBoard(log_dir)]
    self._model_fit(
        strategy,
        callbacks=callbacks,
        use_dataset_creator=use_dataset_creator,
    )
    self.assertTrue(tf.compat.v1.gfile.Exists(log_dir))
    files = tf.compat.v1.gfile.ListDirectory(log_dir)
    self.assertGreaterEqual(len(files), 1)