Skip to content

Commit a002700

Browse files
refactor: renamed csv storing model summary
1 parent 1ed6d34 commit a002700

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/rai_bench/rai_bench/test_models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
get_llm_model_name,
3838
)
3939

40-
REPEATS_SUMMARY_FILE_NAME = "repeats_summary.csv"
40+
MODEL_SUMMARY_FILE_NAME = "model_summary.csv"
4141
TASKS_SUMMARY_FILE_NAME = "tasks_summary.csv"
4242
BENCHMARK_SUMMARY = "benchmark_summary.csv"
4343

@@ -153,7 +153,7 @@ def merge_model_repeats_summary(
153153
repeats=len(summaries),
154154
)
155155

156-
merged_file = model_dir / REPEATS_SUMMARY_FILE_NAME
156+
merged_file = model_dir / MODEL_SUMMARY_FILE_NAME
157157
with open(merged_file, "w", newline="") as f:
158158
writer = csv.DictWriter(f, fieldnames=ModelSummary.model_fields.keys())
159159
writer.writeheader()
@@ -181,7 +181,7 @@ def merge_benchmark_summary(
181181
all_summaries: List[ModelSummary] = []
182182
for model_name in model_names:
183183
model_dir = bench_dir / model_name
184-
merged_file = model_dir / REPEATS_SUMMARY_FILE_NAME
184+
merged_file = model_dir / MODEL_SUMMARY_FILE_NAME
185185

186186
if merged_file.exists():
187187
with open(merged_file, "r") as f:

0 commit comments

Comments
 (0)