tkipf / gym-gridworld

Template for building 2D grid worlds with OpenAI Gym and Pycolab

Gym-Gridworld

This is a simple template with utilities for building a 2D gridworld environment using DeepMind's pycolab and OpenAI gym. This might be useful for re-implementing the environments used in:

CompILE: Compositional Imitation Learning and Execution (ICML 2019)
Thomas Kipf, Yujia Li, Hanjun Dai, Vinicius Zambaldi, Alvaro Sanchez-Gonzalez, Edward Grefenstette, Pushmeet Kohli, Peter Battaglia. https://arxiv.org/abs/1812.01483

Dependencies:

• Python 3.6 or later
• Numpy 1.14 or later
• PyTorch 0.4.1 or later
• OpenAI Gym 0.11 or later