atari-model-zoo
atari-model-zoo copied to clipboard
NotImplementedError when running generate_rollout
Traceback (most recent call last):
File "/home/anand/IDSIA/mutual-information-keypoints/collect_atari_data.py", line 27, in
Same problem. Any workarounds?
It has to do with the API format for the methods like step(), reset(), render() etc. in atari_wrappers.py in atari_zoo which follow the old gym API format of using ._step(). By changing it to align with API format of the latest gym versions (0.15.3) by replacing the "._method_name()" with ".method_name()" and , it worked for me. Hope this helps!
Yeah, thanks, just figured it out too! Instead of updating the model zoo code I went for installing gym==0.12.0 though.
Should now be fixed -- updated code according to @agopal42 's suggestion
@jal278 , thank you!