Skip to content

Commit

Permalink
automatically add test dataset when doing benches
Browse files Browse the repository at this point in the history
  • Loading branch information
winglian committed Oct 10, 2024
1 parent 724f279 commit eba82bf
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions src/axolotl/utils/config/models/input/v0_4_1/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -988,6 +988,20 @@ def check_evals(cls, data):
)
return data

@model_validator(mode="before")
@classmethod
def check_test_datasets_bench(cls, data):
if (
data.get("do_bench_eval")
and not data.get("test_datasets")
and not data.get("val_set_size")
):
LOG.warning(
"`do_bench_eval` needs a test dataset to run evals, adding an empty test_dataset."
)
data["test_datasets"] = [{"path": "axolotl-ai-co/empty-test-ds"}]
return data

@model_validator(mode="before")
@classmethod
def check_eval_packing(cls, data):
Expand Down

0 comments on commit eba82bf

Please sign in to comment.