def __init__(self, env):
    super().__init__(env)
    self.env.reset()
    self.image_shape = self.env.render(mode='rgb_array').shape
    self.q_values = [[0.0, 0.0, 0.0, 0.0]]
def __init__(self, env):
    super().__init__(env)
    self.env.reset()
    self.image_shape = self.env.render(mode='rgb_array').shape
    self.q_values = [[0.0, 0.0, 0.0, 0.0]]