add parameter for output file name

zenotech · Oct 15, 2021 · 16aee26 · 16aee26
1 parent 6081302
commit 16aee26
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 4 deletions.
diff --git a/mycluster/schedulers/base.py b/mycluster/schedulers/base.py
@@ -103,6 +103,7 @@ def create_submit(
         user_email=None,
         qos=None,
         exclusive=True,
+        output_name=None,
     ):
         """
         Write a new job file

diff --git a/mycluster/schedulers/lsf.py b/mycluster/schedulers/lsf.py
@@ -131,6 +131,7 @@ def create_submit(
         user_email=None,
         qos=None,
         exclusive=True,
+        output_name=None,
     ):
         if tasks_per_node is None:
             tasks_per_node = self.tasks_per_node(queue_id)
@@ -147,12 +148,15 @@ def create_submit(
 
         num_queue_slots = num_nodes * tasks_per_node(queue_id)
 
+        if output_name is None:
+            output_name = job_name + ".out"
+
         template = self._load_template("lsf.jinja")
 
         script_str = template.render(
             my_name=job_name,
             my_script=job_script,
-            my_output=job_name,
+            my_output=output_name,
             user_email=user_email,
             queue_name=queue_name,
             num_queue_slots=num_queue_slots,

diff --git a/mycluster/schedulers/pbs.py b/mycluster/schedulers/pbs.py
@@ -153,6 +153,7 @@ def create_submit(
         user_email=None,
         qos=None,
         exclusive=True,
+        output_name=None,
     ):
         queue_name = queue_id
         if tasks_per_node is None:
@@ -178,12 +179,15 @@ def create_submit(
                     tasks_per_node, self._min_tasks_per_node(queue_id)
                 )
 
+        if output_name is None:
+            output_name = job_name + ".out"
+
         template = self._load_template("pbs.jinja")
 
         script_str = template.render(
             my_name=job_name,
             my_script=job_script,
-            my_output=job_name,
+            my_output=output_name,
             user_email=user_email,
             queue_name=queue_name,
             num_queue_slots=num_queue_slots,

diff --git a/mycluster/schedulers/sge.py b/mycluster/schedulers/sge.py
@@ -254,6 +254,7 @@ def create_submit(
         user_email=None,
         qos=None,
         exclusive=True,
+        output_name=None,
     ):
         parallel_env = queue_id.split(":")[0]
         queue_name = queue_id.split(":")[1]
@@ -271,6 +272,9 @@ def create_submit(
         if "mycluster-" in job_script:
             job_script = self._get_data(job_script)
 
+        if output_name is None:
+            output_name = job_name + ".out"
+
         # For exclusive node use total number of slots required
         # is number of nodes x number of slots offer by queue
         num_queue_slots = num_nodes * self.tasks_per_node(queue_id)
@@ -286,7 +290,7 @@ def create_submit(
         script_str = template.render(
             my_name=job_name,
             my_script=job_script,
-            my_output=job_name,
+            my_output=output_name,
             user_email=user_email,
             queue_name=queue_name,
             parallel_env=parallel_env,

diff --git a/mycluster/schedulers/slurm.py b/mycluster/schedulers/slurm.py
@@ -123,6 +123,7 @@ def create_submit(
         user_email=None,
         qos=None,
         exclusive=True,
+        output_name=None,
     ):
         queue_name = queue_id
         if tasks_per_node is None:
@@ -137,12 +138,15 @@ def create_submit(
         if "mycluster-" in job_script:
             job_script = self._get_data(job_script)
 
+        if output_name is None:
+            output_name = job_name + ".out"
+
         template = self._load_template(self._get_template_name())
 
         script_str = template.render(
             my_name=job_name,
             my_script=job_script,
-            my_output=job_name,
+            my_output=output_name,
             user_email=user_email,
             queue_name=queue_name,
             num_tasks=num_tasks,