(self, action)
| 34 | self.action_map = self.env.getActionSet() #[None, 119] |
| 35 | |
| 36 | def step(self, action): |
| 37 | action = self.action_map[action] |
| 38 | reward = self.env.act(action) |
| 39 | done = self.env.game_over() |
| 40 | obs = self.get_observation() |
| 41 | # don't bother returning an info dictionary like gym |
| 42 | return obs, reward, done |
| 43 | |
| 44 | def reset(self): |
| 45 | self.env.reset_game() |