If the slurm job is started with --ntasks_per_node and --nnodes, rather than --ntasks, the env vars currently expected aren' present. Fix that parsing.
Note that I think it'd be good to make the interface to this more general. Maybe all that's needed it to remove "slurm" from the slurm_distributed.py filename and check the possible env vars of different queuing systems (maybe not even actually implement any yet, but at least structure it that way).
If the slurm job is started with --ntasks_per_node and --nnodes, rather than --ntasks, the env vars currently expected aren' present. Fix that parsing.
Note that I think it'd be good to make the interface to this more general. Maybe all that's needed it to remove "slurm" from the
slurm_distributed.pyfilename and check the possible env vars of different queuing systems (maybe not even actually implement any yet, but at least structure it that way).