git clone https://github.com/nacloos/baba-is-ai
cd baba-is-ai
pip install -e .
To play an environment, run this command:
python baba/play.py --env two_room-break_stop-make_winThe --env argument specifies the ID of the environment. Once the game opens, use the arrow keys to move the agent.
You can also create a Gym environment object:
import baba
env_id = "two_room-break_stop-make_win"
env = baba.make(f"env/{env_id}")To list all available environment IDs, run this code:
import baba
print(baba.make("env/*").keys())If you use this project in your research, please cite:
@inproceedings{
cloos2024baba,
title={Baba Is AI: Break the Rules to Beat the Benchmark},
author={Nathan Cloos and Meagan Jens and Michelangelo Naim and Yen-Ling Kuo and Ignacio Cases and Andrei Barbu and Christopher J Cueva},
booktitle={ICML 2024 Workshop on LLMs and Cognition},
year={2024}
}