def METHOD_NAME(
    self, training_callback_attributes: TrainingCallbackAttributes
) -> List[TrainingCallback]:
    callbacks = []
    callbacks.append(
        TrainingCallback(
            where_to_run=[TrainingCallbackLocation.AFTER_TRAIN_ITERATION],
            update_every_num_iters=1,
            func=self.sampler.step_cb,
        )
    )
    return callbacks