Skip to content

Commit 6d65961

Browse files
committed
🎨 Clean up python benchmark parsing script
1 parent eb8b788 commit 6d65961

File tree

3 files changed

+29
-5
lines changed

3 files changed

+29
-5
lines changed

tools/benchmark/gen_map_data.py

100644100755
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
#!/usr/bin/env python3
2+
13
import argparse
24
import random
35

tools/benchmark/parse_bench_data.py

100644100755
Lines changed: 27 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
#!/usr/bin/env python3
2+
3+
import argparse
14
import csv
25
import re
36

@@ -72,9 +75,28 @@ def generate_csv_tables(data, output_prefix):
7275
writer.writerow(row)
7376

7477

75-
# Usage example
76-
file_path = 'bench_output2.txt'
77-
output_prefix = 'output'
78+
def parse_cmdline():
79+
parser = argparse.ArgumentParser()
80+
parser.add_argument(
81+
"--input",
82+
type=str,
83+
required=True,
84+
help=("Path to the input results."),
85+
)
86+
parser.add_argument(
87+
"--output_prefix",
88+
type=str,
89+
required=True,
90+
help="Output filename prefix for the generated CSV file.",
91+
)
92+
return parser.parse_args()
93+
94+
95+
def main():
96+
args = parse_cmdline()
97+
data = parse_file(args.input)
98+
generate_csv_tables(data, args.output_prefix)
99+
78100

79-
data = parse_file(file_path)
80-
generate_csv_tables(data, output_prefix)
101+
if __name__ == "__main__":
102+
main()

tools/gen_release_header.py

100644100755
File mode changed.

0 commit comments

Comments
 (0)