Fix for N/NNodes

C2SM · Jan 29, 2024 · 5a314f4 · 5a314f4
1 parent 7b618ee
commit 5a314f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/config.py b/config.py
@@ -529,7 +529,7 @@ def get_job_info(job_id,
 
     def get_previous_slurm_summary(self,
                                    info_keys=[
-                                       'JobName', 'JobID', 'Partition', 'N',
+                                       'JobName', 'JobID', 'Partition', 'NNodes',
                                        'State', 'Start', 'End', 'Elapsed'
                                    ]):
         """get slurm info summary or all jobs of previous chunk"""
@@ -552,7 +552,7 @@ def print_previous_slurm_summary(self):
             'JobName': 13,
             'JobID': 8,
             'Partition': 9,
-            'N': 3,
+            'NNodes': 3,
             'State': 14,
             'Start': 13,
             'End': 13,
@@ -565,6 +565,7 @@ def print_previous_slurm_summary(self):
         formats = []
         for k in self.info_keys:
             j = info_width[k]
+            k = '#N' if k == 'NNodes' else k
             formats.append(f"{{{k}:>{j}.{j}}}")
             headers.append(f"{k:>{j}.{j}}")
             hlines.append("-" * j)