def METHOD_NAME(self, word_index):
    if word_index % self.stride == 0 and word_index != 0:
        checkpoint_index = word_index - self.expressivity
    else:
        checkpoint_index = (
            math.floor(word_index / self.stride) * self.stride
            + self.stride
            - self.expressivity
        )
    return checkpoint_index