A few minor missing pieces that could be added before merging @lerrytang :
Add procgen to extras_require in main/setup.py
Add testing initialisation of the newly added procgen task in tests/test_init_extra.py (it's in *_extra.py since this task has on the optional dependency.
LGTM overall.
A few minor missing pieces that could be added before merging @lerrytang :