Change set env SLURM_PARTITION to SLURM_JOB_PARTITION

No change in logic, just change name of recently added env var
Formatting fixes

No change in logic, just correct tab/spacing formatting issues