def folder_list_checkpoints()

in src/optimum/nvidia/hub.py [0:0]

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

def folder_list_checkpoints(folder: Path) -> Iterable[Path]:
    checkpoint_candidates = []
    if folder.exists():
        # At this stage we don't know if they are checkpoints or other safetensors files
        re_checkpoint_filename = re.compile(r"rank[0-9]+\.safetensors")
        checkpoint_candidates = list(
            map(
                Path,
                filter(
                    lambda item: re_checkpoint_filename.match(item.name),
                    scandir(folder),
                ),
            )
        )

    return checkpoint_candidates