| 8 | |
| 9 | |
| 10 | class Step: |
| 11 | def __init__(self, state, action, reward, next_state, done): |
| 12 | self.state = state |
| 13 | self.action = action |
| 14 | self.reward = reward |
| 15 | self.next_state = next_state |
| 16 | self.done = done |
| 17 | |
| 18 | |
| 19 | # Transform raw images for input into neural network |