From 6cf82d036f25b7fecf595e535bf74ee249f10f62 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Thu, 24 Feb 2022 17:42:34 +0100 Subject: [PATCH] Add time as 2nd sort criteria --- src/Results.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/src/Results.py b/src/Results.py index eaa729f..aabe7d3 100644 --- a/src/Results.py +++ b/src/Results.py @@ -1,4 +1,5 @@ import os +from operator import itemgetter import math import json import abc @@ -716,13 +717,7 @@ class Summary: data = [x for x in data if x["score"] == score] if model: data = [x for x in data if x["model"] == model] - data = sorted( - data, - key=lambda x: 0.0 - if type(x[sort_key]) is float and math.isnan(x[sort_key]) - else x[sort_key], - reverse=True, - ) + data = sorted(data, key=itemgetter(sort_key, "time"), reverse=True) if number > 0: data = data[:number] max_file = max(len(x["file"]) for x in data)