utils.py 文件源码

def load_grid_selection(path):
    """Load a selected grid from pickle."""
    if not os.path.isfile(path):
        raise ValueError("The file is not existed.""")

    with open(path, "r") as f:
        data = pickle.load(f)
        f.close()

    return data['environment'], data['gt'], data['po'], data['goal']