A minimal environment can be set up by running the following:
conda create -n submission python=3.11.13
conda activate submission
pip install textarena==0.7.3
pip install transformers==4.55.2
pip install torch==2.7.1+cu126 --index-url https://download.pytorch.org/whl/cu126
pip install peft==0.16.0
The original environment is also provided in requirements.txt and submission.yml
A sample of running the agent is provided in offline_play.py
python src/offline_play.py