def _expand_id_suffix()

in submitit/slurm/slurm.py [0:0]

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

def _expand_id_suffix(suffix_parts: str) -> List[str]:
    """Parse the a suffix formatted like "1-3,5,8" into
    the list of numeric values 1,2,3,5,8.
    """
    suffixes = []
    for suffix_part in suffix_parts.split(","):
        if "-" in suffix_part:
            low, high = suffix_part.split("-")
            int_length = len(low)
            for num in range(int(low), int(high) + 1):
                suffixes.append(f"{num:0{int_length}}")
        else:
            suffixes.append(suffix_part)
    return suffixes