Function parse_kv_flags

axlearn/cloud/common/utils.py:120–140

Parses sequence of k:v into a dict. Args: kv_flags: A sequence of strings in the format "k:v". If a key appears twice, the last occurrence "wins". delimiter: The separator between the key and value. Returns: A dict where keys and values are parsed from "

(kv_flags: Sequence[str], *, delimiter: str = ":")

def parse_kv_flags(kv_flags: Sequence[str], *, delimiter: str = ":") -> dict[str, str]:
    """Parses sequence of k:v into a dict.

    Args:
        kv_flags: A sequence of strings in the format "k:v". If a key appears twice, the last
            occurrence "wins".
        delimiter: The separator between the key and value.

    Returns:
        A dict where keys and values are parsed from "k:v".

    Raises:
        ValueError: If a member of `kv_flags` isn't in the format "k:v".
    """
    metadata = {}
    for kv in kv_flags:
        parts = kv.split(delimiter, maxsplit=1)
        if len(parts) != 2:
            raise ValueError(f"Expected key{delimiter}value, got {kv}")
        metadata[parts[0]] = parts[1]
    return metadata


def format_table(*, headings: list[str], rows: list[list[str]]) -> str: