search

Farama-Foundation / Minigrid

Simple and easily configurable grid world environments for reinforcement learning
https://minigrid.farama.org/
Other
2.09k stars 604 forks source link

[Question] Any plan to support BabyAI bot ? #308

Closed GilgameshD closed 1 year ago

GilgameshD commented 1 year ago

Question

Thanks for bringing BabyAI into this repo. I wonder if there is any plan to support the bot that can generate expert demonstrations (link of the script).

pseudo-rnd-thoughts commented 1 year ago

I haven't done much with BabyAI but I would add happy to add it if it is helpful to end users. @rodrigodelazcano Any thoughts? @GilgameshD Would you be interested in adding the file in a PR? It would need to include a copyright statement at the top of the file

rodrigodelazcano commented 1 year ago

I think this could be very valuable in order to generate datasets for Minari. I currently don't have time to work on it but if @GilgameshD you want to contribute to this I'll be happy to help with code review :)

GilgameshD commented 1 year ago

sounds good. I will try to make a PR when I finish it.

Near32 commented 1 year ago

Hello! I have updated the BabyAI bot to be operatable with both the legacy code and the Minigrid code, in case it can speed up full integration in here : https://github.com/mila-iqia/babyai/pull/121 Good luck in the meantime!

GilgameshD commented 1 year ago

thanks @Near32 for providing a reference of the modification. I made the bot.py file compatible with Minigrid and wrote a simple example to run the bot on all BabyAI envs. Where do you think I should put the files? @pseudo-rnd-thoughts @rodrigodelazcano

pseudo-rnd-thoughts commented 1 year ago

@GilgameshD Would you be able to put it in minigrid/utils/bot.py along with several tests in tests/utils/test_bot.py