def generate_image_dict()

in src/script.py [0:0]

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

def generate_image_dict(data_path, slide=None):
    tcga_image_labels = {}
    label = 0
    folders = os.listdir(data_path)
    print(folders)

    for fol in folders:
        files = glob(os.path.join(data_path, fol, f'*{slide}*')) if slide else os.listdir(os.path.join(data_path, fol))
        for file in files:
            path = os.path.join(fol, file)
            tcga_image_labels[path] = label
        label += 1

    return tcga_image_labels