BartekCupial/baba-is-ai

Break the Rules to Beat the Benchmark

Paper

Installation

git clone https://github.com/nacloos/baba-is-ai
cd baba-is-ai
pip install -e .

Usage

To play an environment, run this command:

python baba/play.py --env two_room-break_stop-make_win

The --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())

Citation

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}
}