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
- Python 3.6 or later
- Numpy 1.14 or later
- PyTorch 0.4.1 or later
- OpenAI Gym 0.11 or later