Skip to content

Commit dbaad99

Browse files
committed
add num_samples to results_dict in EvaluationTracker.save()
1 parent 39adf61 commit dbaad99

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
@@ -229,6 +229,7 @@ def save(self) -> None:
229229
"config_tasks": self.task_config_logger.tasks_configs,
230230
"summary_tasks": self.details_logger.compiled_details,
231231
"summary_general": asdict(self.details_logger.compiled_details_over_all_tasks),
232+
"num_samples": self.calculate_num_samples(),
232233
}
233234

234235
# Create the details datasets for later upload

0 commit comments

Comments
 (0)