8f22750dbdac281232241c20736018c47a7a8e26,ants/utils/get_ants_data.py,,get_ants_data,#Any#,10

Before Change

    
    if name == "r16":
        datapath = os.path.join(data_path, "r16slice.jpg")
    elif name == "r27":
        datapath = os.path.join(data_path, "r27slice.jpg")
    elif name == "r64":
        datapath = os.path.join(data_path, "r64slice.jpg")
    elif name == "r85":
        datapath = os.path.join(data_path, "r85slice.jpg")
    elif name == "mni":
        datapath = os.path.join(data_path, "mni.nii.gz")
    elif name == "surf":
        datapath = os.path.join(data_path, "surf.nii.gz")
    else:
        raise ValueError("data file not found")

    return datapath

After Change

    
    datapath = None
    for fname in os.listdir(data_path):
        if name in fname:
            datapath = os.path.join(data_path, fname)

    if datapath is None:
        raise ValueError("File doesnt exist. Options: " , os.listdir(data_path))
    return datapath

Instances