Add in str(task) as a SLURM comment.

markmuetz · Mar 29, 2023 · 2008f7b · 2008f7b
1 parent cb4c5ad
commit 2008f7b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/remake/executor/slurm_executor.py b/remake/executor/slurm_executor.py
@@ -16,6 +16,7 @@
 #SBATCH --job-name={job_name}
 #SBATCH -o {task_slurm_output}/{task_type}_%j.out
 #SBATCH -e {task_slurm_output}/{task_type}_%j.err
+#SBATCH --comment "{comment}"
 {extra_opts}
 {dependencies}
 
@@ -151,10 +152,12 @@ def _write_submit_script(self, task):
                 extra_opts.append(f'#SBATCH --partition={v}')
             else:
                 extra_opts.append(f'#SBATCH --{k}={v}')
+        comment = str(task)
         extra_opts = '\n'.join(extra_opts)
         slurm_script = SLURM_SCRIPT_TPL.format(script_name=script_name,
                                                script_path=script_path,
                                                task_slurm_output=task_slurm_output,
+                                               comment=comment,
                                                remakefile_name=remakefile_name,
                                                remakefile_path=self.remakefile_path,
                                                remakefile_path_hash=self.remakefile_path_hash,