Hi, Does metaflow have any logic if it is in a retry? Ideally I want something like

class TrainFlow(FlowSpec):
    use_latest_check_point = Parameter("use_latest_check_point", default=False)
    @step
    def start(self):
        …
    @retry(times=1)
    @step
    def train(self):
        if self.retry == 1:
            # This step has failed and is now in a retry
            self.use_latest_check_point = True

        train(use_latest_check_point=self.use_latest_check_point)