def METHOD_NAME(self, num_updates):
    """Called at the end of each step."""
    if (
        self.update_schedule is not None
        and num_updates > 0
        and num_updates % self.update_schedule == 0
    ):
        self.step()