Method state

4-atari-hard/2-go-explore.py:283–289

(self)

        self.rooms.update(res["rooms"])

    def state(self):
        return {"cells": {k: {"snapshot": c.snapshot, "score": c.score,
                              "traj_len": c.traj_len, "traj_last": c.traj_last,
                              "seen": c.seen, "chosen": c.chosen,
                              "chosen_since_new": c.chosen_since_new, "lives": c.lives}
                          for k, c in self.cells.items()},
                "rooms": sorted(self.rooms), "done_scores": self.done_scores[-200:]}

    def load_state(self, st):
        self.cells = {}