From 9e9a052bb20810385de1ea1e7c3fb363bc0fc846 Mon Sep 17 00:00:00 2001 From: Guy Gaziv Date: Mon, 24 Oct 2022 15:28:07 -0400 Subject: [PATCH] Fix bug: no returned classes after sorting --- robustness/tools/helpers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/robustness/tools/helpers.py b/robustness/tools/helpers.py index cdc7501..27f6707 100644 --- a/robustness/tools/helpers.py +++ b/robustness/tools/helpers.py @@ -182,7 +182,7 @@ def restricted_label_mapping(classes, class_to_idx, ranges): if idx in range_set: mapping[class_name] = new_idx # assert class_name in mapping - filtered_classes = list(mapping.keys()).sort() + filtered_classes = sorted(list(mapping.keys())) return filtered_classes, mapping def custom_label_mapping(classes, class_to_idx, ranges): @@ -193,5 +193,5 @@ def custom_label_mapping(classes, class_to_idx, ranges): if idx in range_set: mapping[class_name] = new_idx - filtered_classes = list(mapping.keys()).sort() + filtered_classes = sorted(list(mapping.keys())) return filtered_classes, mapping