#!/bin/sh
set -e

# # download mjpro150
# mkdir /root/.mujoco
# cd /root/.mujoco
# wget https://www.roboti.us/download/mjpro150_linux.zip
# unzip mjpro150_linux.zip

# copy the mujoco license key into the container
# echo $MUJOCO_KEY | base64 --decode > /root/.mujoco/mjkey.txt
# export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/root/.mujoco/mjpro150/bin

# git clone https://github.com/deepmind/dm_control.git
# pip3 install ./dm_control

export VIZDOOM_ROOT=`pip show vizdoom 2>/dev/null | awk '/Location/{print $2}'`/vizdoom

bash -c "$@"