Skip to content

Commit 6bfdae6

Browse files
sayakpaulMcPatate
andauthored
Apply suggestions from code review
Co-authored-by: Luc Georges <[email protected]>
1 parent 858dc09 commit 6bfdae6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

benchmarks/populate_into_db.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99

1010
FINAL_CSV_FILENAME = "collated_results.csv"
1111
# https://github.com/huggingface/transformers/blob/593e29c5e2a9b17baec010e8dc7c1431fed6e841/benchmark/init_db.sql#L27
12-
TABLE_NAME = "model_measurements"
12+
BENCHMARKS_TABLE_NAME = "benchmarks"
13+
MEASUREMENTS_TABLE_NAME = "model_measurements"
1314

1415
if __name__ == "__main__":
1516
try:
@@ -98,7 +99,7 @@ def _cast_value(val, dtype: str):
9899

99100
# Batch-insert all rows
100101
insert_sql = f"""
101-
INSERT INTO {TABLE_NAME} (
102+
INSERT INTO {MEASUREMENTS_TABLE_NAME} (
102103
benchmark_id,
103104
measurements
104105
)

0 commit comments

Comments
 (0)