diff --git a/R/clusterFunctionsSlurm.R b/R/clusterFunctionsSlurm.R index 321fea6d..b836e45a 100644 --- a/R/clusterFunctionsSlurm.R +++ b/R/clusterFunctionsSlurm.R @@ -92,19 +92,30 @@ makeClusterFunctionsSlurm = function(template = "slurm", clusters = NULL, array. } listJobsQueued = function(reg) { - args = c("-h", "-o %i", "-u $USER", "-t PD", sprintf("--clusters=%s", clusters)) + if (is.null(clusters) && !is.null(reg$default.resources$clusters)) + clusters = reg$default.resources$clusters + args = c("-h", "-o %i", "-u $USER", "-t PD", + sprintf("--clusters=%s", clusters), + sprintf("--partition=%s", reg$default.resources$partition)) listJobs(reg, args) } listJobsRunning = function(reg) { - args = c("-h", "-o %i", "-u $USER", "-t R,S,CG", sprintf("--clusters=%s", clusters)) + if (is.null(clusters) && !is.null(reg$default.resources$clusters)) + clusters = reg$default.resources$clusters + args = c("-h", "-o %i", "-u $USER", "-t R,S,CG", + sprintf("--clusters=%s", clusters), + sprintf("--partition=%s", reg$default.resources$partition)) listJobs(reg, args) } killJob = function(reg, batch.id) { assertRegistry(reg, writeable = TRUE) assertString(batch.id) - cfKillJob(reg, "scancel", c(sprintf("--clusters=%s", clusters), batch.id), nodename = nodename) + if (is.null(clusters) && !is.null(reg$default.resources$clusters)) + clusters = reg$default.resources$clusters + cfKillJob(reg, "scancel", c(sprintf("--clusters=%s", clusters), + sprintf("--partition=%s", reg$default.resources$partition), batch.id), nodename = nodename) } makeClusterFunctions(name = "Slurm", submitJob = submitJob, killJob = killJob, listJobsRunning = listJobsRunning,