Wait for rslurm::slurm_apply to complete

wait_for_slurm.r

# Ben Fasoli
# Wait for rslurm::slurm_apply to complete

library(rslurm)

fun <- function(x) {
  Sys.sleep(x)
  return(T)
}

slurm_options <- list(
  time      = '300:00:00',
  account   = 'lin-kp',
  partition = 'lin-kp'
)

output <- slurm_apply(fun, data.frame(x = c(5, 10)), nodes = 1, 
                      cpus_per_node = 2, slurm_options = slurm_options)

wait_for_slurm <- function(slr_job) {
  cmd <- paste('squeue -hn', slr_job$jobname)
  while (length(system(cmd, intern = T)) > 0) {
    Sys.sleep(1)
  }
}

wait_for_slurm(output)