From e2c295a05a686aeba1dab41b814d538059b7bbef Mon Sep 17 00:00:00 2001 From: Patrick Gartung Date: Thu, 19 Sep 2024 15:31:52 +0200 Subject: [PATCH] Account for missing step5 in some workflows --- comparisons/compare-maxmem-summary.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/comparisons/compare-maxmem-summary.py b/comparisons/compare-maxmem-summary.py index 53fa5619ace..b48ea9312c5 100755 --- a/comparisons/compare-maxmem-summary.py +++ b/comparisons/compare-maxmem-summary.py @@ -27,7 +27,6 @@ def compare_maxmem_summary(**kwargs): inputFiles = glob.glob(os.path.join(inputDir, filePattern)) - steps = ["step1", "step2", "step3", "step4", "step5"] workflows = {} for inputFile in sorted(inputFiles): with open(inputFile, "r") as f: @@ -39,7 +38,7 @@ def compare_maxmem_summary(**kwargs): threshold = float(jsonDict["threshold"]) if workflow not in workflows: workflows[workflow] = {} - for step in steps: + for step in max_memory_pr_dict.keys(): max_mem_pr = max_memory_pr_dict[step].get("max memory used") if max_mem_pr: max_memory_pr = max_mem_pr / 1000000 @@ -86,7 +85,7 @@ def compare_maxmem_summary(**kwargs): max_mem_pr, max_mem_base, max_mem_pdiff = 0, 0, 0 for workflow in sortedworkflows: - for step in steps: + for step in sorted(workflows[workflow].keys()): max_mem_pr = workflows[workflow][step]["max memory pr"] max_mem_base = workflows[workflow][step]["max memory base"] max_mem_pdiff = workflows[workflow][step]["max memory pdiffs"]