#!/bin/bash --login
# Sets up extra dependencies for Unity

# Fail on the first error; killable by SIGINT
set -e
trap "exit" INT

echo "--- Installing Unity ML agents ---"
conda activate lab
pip install gym_unity==0.4.5
# clone to slm_lab/env/SLM-Env
git clone https://github.com/kengz/SLM-Env.git ./slm_lab/env/SLM-Env

echo "--- Installing VizDoom ---"
pip install vizdoom==1.1.6
