Updated for new pyslurm classes