human_feedback.py 文件源码

def save_labels(filename, episode, frames):
    """
    Args:
        filename (str): file to save labeled episode to.
        episode (episode): an episode to save labeled `frames` to (this is pickled and gzipped)
        frames (list of frames): a time ordered list of labeled frames.

    Summary:
        Save labeled frames to a file.
    """

    episode.frames = frames
    with gzip.open(filename, "wb") as f:
        pickler = pickle.Pickler(f)
        pickler.dump(episode)