5ac177662e054b9b8f4ad95fc9053561cc69a6a8,floydker/src/floydker/utils.py,,gen_target_env_from_tag,#Any#,124

Before Change

    sample input: "tensorflow:1.0.1-gpu-py3"
    sample output: ("1.0.1", "py3.gpu")
    
    img_commit = img_tag.split(":")[-1]
    version, target = img_commit.split("-", 1)
    target = ".".join(reversed(target.split("-", 1)))
    return version, target

After Change

    sample input: "tensorflow:1.0.1-gpu-py3_aws"
    sample output: ("1.0.1", "py3.gpu_aws")
    
    match = docker_tag_re.match(img_tag)
    if not match:
        return None, None
    target_env = match.group("env")
    if match.group("arch"):
        target_env += "." + match.group("arch")
    if match.group("cloud"):
        target_env += "_" + match.group("cloud")
    return match.group("version"), target_env

Instances