Skip to content

Commit 39adf61

Browse files
committed
add num_samples to final_dict of evaluation_tracker
1 parent 2869681 commit 39adf61

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/lighteval/logging/evaluation_tracker.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -352,6 +352,7 @@ def generate_final_dict(self) -> dict:
352352
"config_tasks": self.task_config_logger.tasks_configs,
353353
"summary_tasks": self.details_logger.compiled_details,
354354
"summary_general": asdict(self.details_logger.compiled_details_over_all_tasks),
355+
"num_samples": self.calculate_num_samples(),
355356
}
356357

357358
final_dict = {

0 commit comments

Comments
 (0)