def get_resume_file()

in main.py [0:0]

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

def get_resume_file(filename):
    if isfile(filename):
        return filename
    filelist = glob.glob(os.path.join(params.checkpoint_dir, '*.tar'))
    if len(filelist) == 0:
        return None

    epochs = np.array([int(os.path.splitext(os.path.basename(x))[0]) for x in filelist])
    max_epoch = np.max(epochs)
    resume_file = os.path.join(params.checkpoint_dir, '{:d}.tar'.format(max_epoch))
    return resume_file