Skip to content

Commit 5fc5849

Browse files
committed
makes neater for loop in print_summary
1 parent 663cde1 commit 5fc5849

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

benchmarks/utils.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -42,25 +42,21 @@ def print_summary(results):
4242

4343
print(f"\n------ Summary of Case {count_case:02}: {cases[count_case]}: ")
4444
L = []
45-
timings = []
4645
time_min = []
4746
time_max = []
4847
time_avg = []
4948
benchmark_pairs = results[cases[count_case]]
5049
number_of_benchmarks = len(benchmark_pairs)
5150
print(f"Number of benchmarks = {number_of_benchmarks:04} ")
5251

53-
parameter_tuple_list = list(benchmark_pairs.keys())
54-
for count_benchmark in range(number_of_benchmarks):
55-
L.append(parameter_tuple_list[count_benchmark][0][1])
56-
benchmarked_values = benchmark_pairs[parameter_tuple_list[count_benchmark]]
57-
timings.append(benchmarked_values["time"])
58-
time_min.append(min(timings[-1]))
59-
time_max.append(max(timings[-1]))
60-
time_avg.append(sum(timings[-1]) / len(timings[-1]))
52+
for parameter_tuple in benchmark_pairs.keys():
53+
L.append(parameter_tuple[0][1])
54+
times = benchmark_pairs[parameter_tuple]["time"]
55+
time_min.append(min(times))
56+
time_max.append(max(times))
57+
time_avg.append(sum(times) / len(times))
6158

6259
summary[cases[count_case]]["L"] = L
63-
summary[cases[count_case]]["timings"] = timings
6460
summary[cases[count_case]]["time_min"] = time_min
6561
summary[cases[count_case]]["time_max"] = time_max
6662
summary[cases[count_case]]["time_avg"] = time_avg

0 commit comments

Comments
 (0)