def dump_images()

in datasets/lsun_bedroom.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def dump_images(out_dir, images, prefix):
    if not os.path.exists(out_dir):
        os.mkdir(out_dir)
    for i, img in enumerate(images):
        Image.fromarray(img).save(os.path.join(out_dir, f"{prefix}_{i:07d}.png"))