Skip to content

Commit b026810

Browse files
committed
change to list comprehension
1 parent 93132dc commit b026810

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/test_evaluations/test_evaluation_functions.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -128,12 +128,8 @@ def test_evaluation_list_sort(self):
128128
"predictive_accuracy", size=size, offset=0, task=[task_id], sort_order="desc")
129129
self.assertEqual(len(sorted_eval), size)
130130
self.assertGreater(len(unsorted_eval), 0)
131-
sorted_output = []
132-
unsorted_output = []
133-
for eval in sorted_eval.values():
134-
sorted_output.append(eval.value)
135-
for eval in unsorted_eval.values():
136-
unsorted_output.append(eval.value)
131+
sorted_output = [evaluation.value for evaluation in sorted_eval.values()]
132+
unsorted_output = [evaluation.value for evaluation in unsorted_eval.values()]
137133

138134
# Check if output from sort is sorted in the right order
139135
self.assertTrue(sorted(sorted_output, reverse=True) == sorted_output)

0 commit comments

Comments
 (0)