-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlsf.py
25 lines (21 loc) · 927 Bytes
/
lsf.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import luigi
class LSFJobTask(luigi.Task):
"""
Takes care of uploading and executing an LSF job
"""
n_cpu_flag = luigi.IntParameter(default=2, significant=False)
shared_tmp_dir = luigi.Parameter(default='/tmp', significant=False)
resource_flag = luigi.Parameter(default='mem=8192', significant=False)
memory_flag = luigi.Parameter(default='8192', significant=False)
queue_flag = luigi.Parameter(default='queue_name', significant=False)
runtime_flag = luigi.IntParameter(default=60)
job_name_flag = luigi.Parameter(default='')
poll_time = luigi.FloatParameter(
significant=False, default=5,
description="specify the wait time to poll bjobs for the job status")
save_job_info = luigi.BoolParameter(default=False)
output = luigi.Parameter(default='')
extra_bsub_args = luigi.Parameter(default='')
job_status = None
def work(self):
return