Setting a nice parameter (the job priority).

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/clusters.py

@@ -4,7 +4,7 @@
 __all__ = ["slurm_lines", "rc_lines"]
 
 
-def slurm_lines(queue, job_name, array=None, mem=None, out_name=None):
+def slurm_lines(queue, job_name, array=None, mem=None, out_name=None, nice_parameter=None):
     """
     Function for creating the general lines that slurm scripts are starting with.
 
@@ -20,6 +20,8 @@ def slurm_lines(queue, job_name, array=None, mem=None, out_name=None):
         Requested memory. If None cluster default (5 GB) will be used
     out_name : None or str
         If the output should be written to a specific output file
+    nice_parameter : None or int
+        Job priority
 
     Returns
     -------
@@ -35,6 +37,7 @@ def slurm_lines(queue, job_name, array=None, mem=None, out_name=None):
         "#SBATCH -n 1\n\n",
         f"#SBATCH --output={out_name}.out\n" if out_name is not None else "",
         f"#SBATCH --error={out_name}.err\n\n" if out_name is not None else "",
+        f"#SBATCH --nice={nice_parameter}\n"
         "ulimit -l unlimited\n",
         "ulimit -s unlimited\n",
         "ulimit -a\n\n",

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/coincident_events.py

@@ -73,7 +73,7 @@ def configfile_coincidence(target_dir, source_name, config_file):
         yaml.dump(conf, f, default_flow_style=False)
 
 
-def linking_bash_lst(target_dir, LST_runs, source_name, LST_version, env_name, cluster):
+def linking_bash_lst(target_dir, LST_runs, source_name, LST_version, env_name, cluster, Nice_parameter):
 
     """
     This function links the LST data paths to the working directory and creates bash scripts.
@@ -92,6 +92,8 @@ def linking_bash_lst(target_dir, LST_runs, source_name, LST_version, env_name, c
         Name of the conda environment
     cluster : str
         Cluster system
+    Nice_parameter : int
+        Job priority
     """
 
     coincidence_DL1_dir = f"{target_dir}/v{__version__}/{source_name}"
@@ -143,6 +145,7 @@ def linking_bash_lst(target_dir, LST_runs, source_name, LST_version, env_name, c
                     array=process_size,
                     mem="6g",
                     out_name=f"{outputdir}/logs/slurm-%x.%A_%a",
+                    nice_parameter=Nice_parameter,
                 )
                 rc = rc_lines(
                     store="$SAMPLE ${SLURM_ARRAY_JOB_ID} ${SLURM_ARRAY_TASK_ID}",
@@ -194,6 +197,7 @@ def main():
     env_name = config["general"]["env_name"]
     LST_version = config["general"]["LST_version"]
     config_file = config["general"]["base_config_file"]
+    nice_parameter = config["general"]["nice"]
 
     source_in = config["data_selection"]["source_name_database"]
     source = config["data_selection"]["source_name_output"]
@@ -227,6 +231,7 @@ def main():
                 LST_version,
                 env_name,
                 cluster,
+                nice_parameter,
             )  # linking the data paths to current working directory
 
             print("***** Submitting processess to the cluster...")

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/config_auto_MCP.yaml

@@ -1,3 +1,11 @@
+mc_tel_ids:
+    LST-1: 1
+    LST-2: 0
+    LST-3: 0
+    LST-4: 0
+    MAGIC-I: 2
+    MAGIC-II: 3
+
 directories:
     workspace_dir: "/fefs/aswg/workspace/elisa.visentin/auto_MCP_PR/"  # Output directory where all the data products will be saved.
 
@@ -21,4 +29,6 @@ general:
     nsb: [0.5, 1.0, 1.5, 2.0, 2.5, 3.0]
     env_name: magic-lst  # name of the conda environment to be used to process data.
     cluster: "SLURM"  # cluster management system on which data are processed. At the moment we have only SLURM available, in the future maybe also condor (PIC, CNAF).
+    nice: 0 # Set the job priority >0. A lower nice value increases the priority, while a higher value reduces it. Default value is equal 0. 
+
 

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/dl1_production.py

@@ -75,7 +75,7 @@ def config_file_gen(target_dir, source_name, config_file):
 
 
 def lists_and_bash_gen_MAGIC(
-    target_dir, telescope_ids, MAGIC_runs, source, env_name, cluster
+    target_dir, telescope_ids, MAGIC_runs, source, env_name, cluster, Nice_parameter
 ):
 
     """
@@ -95,6 +95,8 @@ def lists_and_bash_gen_MAGIC(
         Name of the environment
     cluster : str
         Cluster system
+    Nice_parameter : int
+        Job priority
     """
     if cluster != "SLURM":
         logger.warning(
@@ -106,6 +108,7 @@ def lists_and_bash_gen_MAGIC(
         queue="short",
         job_name=process_name,
         out_name=f"{target_dir}/v{__version__}/{source}/DL1/slurm-linkMAGIC-%x.%j",
+        nice_parameter=Nice_parameter,
     )
 
     with open(f"{source}_linking_MAGIC_data_paths.sh", "w") as f:
@@ -137,6 +140,7 @@ def lists_and_bash_gen_MAGIC(
                     array=number_of_nodes,
                     mem="2g",
                     out_name=f"{target_dir}/v{__version__}/{source}/DL1/M{magic}/{i[0]}/{i[1]}/logs/slurm-%x.%A_%a",
+                    nice_parameter=Nice_parameter,
                 )
                 rc = rc_lines(
                     store="$SAMPLE ${SLURM_ARRAY_JOB_ID} ${SLURM_ARRAY_TASK_ID}",
@@ -220,6 +224,7 @@ def main():
     source = config["data_selection"]["source_name_output"]
     cluster = config["general"]["cluster"]
     target_dir = Path(config["directories"]["workspace_dir"])
+    nice_parameter = config["general"]["nice"]
 
     if source_in is None:
         source_list = joblib.load("list_sources.dat")
@@ -255,6 +260,7 @@ def main():
             source_name,
             env_name,
             cluster,
+            nice_parameter,
         )  # MAGIC real data
         if (telescope_ids[-2] > 0) or (telescope_ids[-1] > 0):
             list_of_MAGIC_runs = glob.glob(f"{source_name}_MAGIC-*.sh")

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/merge_stereo.py

@@ -27,7 +27,7 @@
 logger.setLevel(logging.INFO)
 
 
-def MergeStereo(target_dir, env_name, source, cluster):
+def MergeStereo(target_dir, env_name, source, cluster, Nice_parameter):
     """
     This function creates the bash scripts to run merge_hdf_files.py in all DL1Stereo subruns.
 
@@ -41,6 +41,8 @@ def MergeStereo(target_dir, env_name, source, cluster):
         Name of the target
     cluster : str
         Cluster system
+    Nice_parameter : int
+        Job priority
     """
 
     process_name = source
@@ -64,6 +66,7 @@ def MergeStereo(target_dir, env_name, source, cluster):
             job_name=f"{process_name}_stereo_merge",
             mem="2g",
             out_name=f"{stereoMergeDir}/logs/slurm-%x.%A_%a",
+            nice_parameter=Nice_parameter,
         )
         rc = rc_lines(
             store=f"{nightLST} ${{SLURM_JOB_ID}}", out=f"{stereoMergeDir}/logs/list"
@@ -109,6 +112,7 @@ def main():
     source_in = config["data_selection"]["source_name_database"]
     source = config["data_selection"]["source_name_output"]
     cluster = config["general"]["cluster"]
+    nice_parameter = config["general"]["nice"]
 
     if source_in is None:
         source_list = joblib.load("list_sources.dat")
@@ -121,7 +125,7 @@ def main():
     for source_name in source_list:
 
         print("***** Merging DL1Stereo files run-wise...")
-        MergeStereo(target_dir, env_name, source_name, cluster)
+        MergeStereo(target_dir, env_name, source_name, cluster, nice_parameter)
 
         list_of_merge = glob.glob(f"{source_name}_StereoMerge_*.sh")
         if len(list_of_merge) < 1:

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/merging_runs.py

@@ -33,7 +33,7 @@
 logger.setLevel(logging.INFO)
 
 
-def merge(target_dir, MAGIC_runs, env_name, source, cluster):
+def merge(target_dir, MAGIC_runs, env_name, source, cluster, Nice_parameter):
 
     """
     This function creates the bash scripts to run merge_hdf_files.py for real data
@@ -50,6 +50,8 @@ def merge(target_dir, MAGIC_runs, env_name, source, cluster):
         Target name
     cluster : str
         Cluster system
+    Nice_parameter : int
+        Job priority
     """
 
     process_name = f"merging_{source}"
@@ -66,6 +68,7 @@ def merge(target_dir, MAGIC_runs, env_name, source, cluster):
         job_name=process_name,
         mem="2g",
         out_name=f"{MAGIC_DL1_dir}/Merged/logs/slurm-%x.%j",
+        nice_parameter=Nice_parameter,
     )
     os.makedirs(f"{MAGIC_DL1_dir}/Merged/logs", exist_ok=True)
 
@@ -120,6 +123,7 @@ def main():
     source_in = config["data_selection"]["source_name_database"]
     source = config["data_selection"]["source_name_output"]
     cluster = config["general"]["cluster"]
+    nice_parameter = config["general"]["nice"]
 
     if source_in is None:
         source_list = joblib.load("list_sources.dat")
@@ -144,6 +148,7 @@ def main():
             env_name,
             source_name,
             cluster,
+            nice_parameter,
         )  # generating the bash script to merge the subruns
 
         print("***** Running merge_hdf_files.py on the MAGIC data files...")

magicctapipe/scripts/lst1_magic/semi_automatic_scripts/stereo_events.py

@@ -63,7 +63,7 @@ def configfile_stereo(target_dir, source_name, config_file):
         yaml.dump(conf, f, default_flow_style=False)
 
 
-def bash_stereo(target_dir, source, env_name, cluster):
+def bash_stereo(target_dir, source, env_name, cluster, Nice_parameter):
 
     """
     This function generates the bashscripts for running the stereo analysis.
@@ -78,6 +78,8 @@ def bash_stereo(target_dir, source, env_name, cluster):
         Name of the environment
     cluster : str
         Cluster system
+    Nice_parameter : int
+        Job priority
     """
 
     process_name = source
@@ -114,6 +116,7 @@ def bash_stereo(target_dir, source, env_name, cluster):
             array=process_size,
             mem="2g",
             out_name=f"{stereoDir}/logs/slurm-%x.%A_%a",
+            nice_parameter=Nice_parameter,
         )
         rc = rc_lines(
             store="$SAMPLE ${SLURM_ARRAY_JOB_ID} ${SLURM_ARRAY_TASK_ID}",
@@ -166,6 +169,7 @@ def main():
     source = config["data_selection"]["source_name_output"]
 
     cluster = config["general"]["cluster"]
+    nice_parameter = config["general"]["nice"]
 
     if source_in is None:
         source_list = joblib.load("list_sources.dat")
@@ -182,7 +186,7 @@ def main():
         # Below we run the analysis on the real data
 
         print("***** Generating the bashscript...")
-        bash_stereo(target_dir, source_name, env_name, cluster)
+        bash_stereo(target_dir, source_name, env_name, cluster, nice_parameter)
 
         print("***** Submitting processess to the cluster...")
         print(f"Process name: {source_name}_stereo")