From 5a314f4b659bd24f9568676ade3148bce66bd1fc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20J=C3=A4hn?= <mjaehn@ethz.ch>
Date: Mon, 29 Jan 2024 12:04:30 +0100
Subject: [PATCH] Fix for N/NNodes

---
 config.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/config.py b/config.py
index 4f11c61b..d5517817 100644
--- 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)