diff --git a/src/scgenerator/data/submit_job_template.txt b/src/scgenerator/data/submit_job_template.txt index 397841a..19842e3 100644 --- a/src/scgenerator/data/submit_job_template.txt +++ b/src/scgenerator/data/submit_job_template.txt @@ -9,7 +9,7 @@ ### Give all resources to a single Ray task, ray can manage the resources internally #SBATCH --ntasks-per-node=1 #SBATCH --cpus-per-task={cpus_per_node} - +{dependency} # Load modules or your own conda environment here {environment_setup} diff --git a/src/scgenerator/scripts/slurm_submit.py b/src/scgenerator/scripts/slurm_submit.py index 70e05a8..5ca8092 100644 --- a/src/scgenerator/scripts/slurm_submit.py +++ b/src/scgenerator/scripts/slurm_submit.py @@ -92,6 +92,7 @@ def create_parser(): parser.add_argument( "--command", default="run", choices=["run", "resume", "merge"], help="command to run" ) + parser.add_argument("--dependency", default=None, help="sbatch dependency argument") return parser @@ -112,6 +113,11 @@ def main(): template = Paths.gets("submit_job_template") args = parser.parse_args() + if args.dependency is None: + args.dependency = "" + else: + args.dependency = f"#SBATCH --dependency={args.dependency}" + if not re.match(r"^[0-9]{2}:[0-9]{2}:[0-9]{2}$", args.time) and not re.match( r"^[0-9]+$", args.time ):