#!/bin/bash
#SBATCH -N 1
#SBATCH -C gpu
#SBATCH -q shared
#SBATCH -t 06:00:00
#SBATCH -A atlas
#SBATCH -o jobs/slurm/%j.out # STDOUT

supervisor="$1"
coder="$2"
NUM_TESTS="$3"
OUTDIR="$4"

module load python
source ~/.bashrc
conda activate llm_env

python jobs/test_models.py "$supervisor" "$coder" "$NUM_TESTS" --outdir "$OUTDIR"