def get_checkpoint()

in src/alignment/model_utils.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def get_checkpoint(training_args: SFTConfig | DPOConfig) -> Path | None:
    last_checkpoint = None
    if os.path.isdir(training_args.output_dir):
        last_checkpoint = get_last_checkpoint(training_args.output_dir)
    return last_checkpoint