#!/bin/bash

export PYTHONPATH=.

algos=(lka_v2)
suite=mpe
# envs=(escalation staghunt harvest)
# envs=(staghunt escalation)
envs=(simple_spread)
# config=magw
config=mpe
# args1=(uni sim)
args1=(sim)
# args2=(3 5 8)
# args2=(3 5)
args2=(3)
args3=(5e-4)
date=$(date +"%m%d")
na=3
info=""
ld="${suite}-logs"
gpus=$(nvidia-smi -L | wc -l)
# for x in $(seq 1 1); do
#     shift
# done

if [ -z "$info" ]; then
    n=$date
    echo $n
else
    n="$date-$info"
    echo $n
fi

i=0
n=debug
for e in "${envs[@]}"; do
    config=$e
    for a in "${algos[@]}"; do
        for a1 in "${args1[@]}"; do
            for a2 in "${args2[@]}"; do
                s=0
                kws="imaginary_rollout=$a1 n_imaginary_runs=$a2"
                env="${suite}-${e}"
                if [[ "$OSTYPE" == 'linux-gnu'* ]]; then
                    gpu=$(($i % $gpus))
                    py_script="python run/train.py -a ${a} -e $env -c $config -ld $ld -n $n -s $s --gpu $gpu -na $na -kw $kws &"
                else
                    py_script="python run/train.py -a ${a} -e $env -c $config -ld $ld -n $n -s $s -na $na -kw $kws &"
                fi
                echo $py_script
                eval $py_script
                ((i=(i+1) ))
            done
        done
        wait
    done
    wait
done
wait
echo "Script completed"
