Update slurmmanager.jl

Author: DilumAluthge

src/slurmmanager.jl

@@ -20,10 +20,8 @@ mutable struct SlurmManager <: ClusterManager
 end
 
 function get_slurm_ntasks_int()
-  if "SLURM_JOB_ID" in keys(ENV)
+  if haskey(ENV, "SLURM_NTASKS")
     ntasks_str = ENV["SLURM_JOB_ID"]
-  elseif "SLURM_JOBID" in keys(ENV)
-    ntasks_str = ENV["SLURM_JOBID"]
   else
     msg = "SlurmManager must be constructed inside a Slurm allocation environment." *
           "SLURM_JOB_ID or SLURM_JOBID must be defined."
@@ -34,9 +32,9 @@ function get_slurm_ntasks_int()
 end
 
 function get_slurm_jobid_int()
-  if "SLURM_JOB_ID" in keys(ENV)
+  if haskey(ENV, "SLURM_JOB_ID")
       jobid_str = ENV["SLURM_JOB_ID"]
-  elseif "SLURM_JOBID" in keys(ENV)
+  elseif haskey(ENV, "SLURM_JOBID")
       jobid_str = ENV["SLURM_JOBID"]
   else
       msg = "SlurmManager must be constructed inside a Slurm allocation environment." *