#!/bin/bash
#SBATCH --job-name=dppo_d3il_install
#SBATCH --account=hk-project-p0022232
#SBATCH --partition=dev_accelerated
#SBATCH --gres=gpu:1
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=4
#SBATCH --time=00:20:00
#SBATCH --mem=16G
#SBATCH --output=logs/dppo_d3il_install_%j.out
#SBATCH --error=logs/dppo_d3il_install_%j.err

module load devel/cuda/12.4

echo "Installing D3IL for DPPO..."
cd $SLURM_SUBMIT_DIR
source .venv/bin/activate

# Install D3IL fork
echo "Cloning D3IL repository..."
git clone https://github.com/allenzren/d3il d3il_repo

cd d3il_repo
echo "Installing d3il environments..."
pip install -e environments/d3il
pip install -e environments/d3il/envs/gym_avoiding_env/

echo "D3IL installation completed!"
cd $SLURM_SUBMIT_DIR

# Test import
echo "Testing D3IL import..."
python -c "import d3il_benchmark; print('D3IL imported successfully!')"