def METHOD_NAME(self, strategy, use_dataset_creator):
    class MyCallback(callbacks_lib.Callback):
        pass
    my_callback = MyCallback()
    callbacks = [my_callback]
    self._model_fit(
        strategy,
        callbacks=callbacks,
        use_dataset_creator=use_dataset_creator,
    )
    # PSStrategy should default to epoch-level logging.
    self.assertEqual(my_callback.params["verbose"], 2)