def reset(self):
  self.game.restart()
  self.state = self.game.get_state()
  self.game.pause()
  return self.state.snapshot