def copy_py()

in run_with_submitit.py [0:0]


def copy_py(dst_folder, root='.'):
    if not os.path.exists(dst_folder):
        print("Folder doesn't exist!")
        return
    for f in os.listdir(root):
        if f.endswith('.py'):
            shutil.copy2(f, dst_folder)