def step(self) -> float:
        """
        Grow the value by self.step_per_epoch.

        :return: The updated scheduler value.
        """
        self.step_count += 1
        if self.step_count > self.warmup and self.val < self.final_val:
            self.val += self.step_per_epoch
            self.val = min(self.val, self.final_val)
        return self.val

        Grow the value by self.step_per_epoch.

        :return: The updated scheduler value.
        
art/defences/trainer/ibp_certified_trainer_pytorch.py