Allow specifiying a partion to submit slurm jobs