def save_codec_checkpoint()

in training/utils.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def save_codec_checkpoint(output_dir, dataset, step):
    checkpoint_path = f"{CHECKPOINT_CODEC_PREFIX}-{step}"
    output_path = os.path.join(output_dir, checkpoint_path)
    dataset.save_to_disk(output_path)