DQN throws unexpected keyword argument 'dim'

[Phantomb]

On the latest version, I get the following error:

> python main.py dqn single_room --double_dqn --normalize_reward --malmo_ports 10001
2018-01-16 15:57:36 INFO: To view results, run 'python -m visdom.server'
2018-01-16 15:57:36 INFO: then head over to http://localhost:8097
Traceback (most recent call last):
  File "main.py", line 79, in <module>
    True)
  File "C:\Github_Repos\tessler_Malmo\utilities\helpers.py", line 52, in play_full_episode
    action = policy.get_action(states, is_train)
  File "C:\Github_Repos\tessler_Malmo\policies\dqn.py", line 125, in get_action
    self.previous_states = np.expand_dims(self.previous_states, dim=0)
TypeError: expand_dims() got an unexpected keyword argument 'dim'

The relevant commit is 9866be5.

Did you mean to use 'axis' in place of dim?

[tesslerc]

Thanks! I have been using retain_rgb for a while, which is probably why I missed this.
d2a30f5

Tell me if you still encounter any issues 👍

[tesslerc]

Assuming fixed.

[Phantomb]

Yes it seems fixed. I initially missed your reply to this, apologies.