Get BabyAI Datasets Working

[eihli]

Part of a larger task to acquire as many of the GATO control datasets as we can (listed in the image below).

Our existing control datasets come from D4RL which we convert to Minari

I provide the above for context. The task for this issue is the BabyAI dataset, which is a bit different.

https://github.com/snat-s/baby-ai-dataset/blob/master/scripts/bot.py

This was originaly a test but it can be used to generate a dataset.
The original is from here: https://github.com/Farama-Foundation/Minigrid/blob/master/tests/test_baby_ai_bot.py

Then I used this as inspiration to generate datasets for each level: https://github.com/rodrigodelazcano/d4rl-minari-dataset-generation/blob/main/scripts/minigrid/generate_minigrid.py

To run this you are going to need both, minari and minigrid cloned because I
am using stuff that has not shipped to the packages.