Skip to content

Commit b844272

Browse files
committed
added v2 task
1 parent 8647315 commit b844272

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

src/inspect_evals/sec_qa/sec_qa.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,26 @@ def sec_qa_v1() -> Task:
3838
)
3939

4040

41+
@task
42+
def sec_qa_v2() -> Task:
43+
"""Inspect Task implementing the SecQA benchmark v2"""
44+
# dataset
45+
dataset = hf_dataset(
46+
path="zefang-liu/secqa",
47+
name="secqa_v2", # aka subset
48+
split="dev",
49+
sample_fields=record_to_sample,
50+
trust=True,
51+
)
52+
53+
# define task
54+
return Task(
55+
dataset=dataset,
56+
solver=[system_message(SYSTEM_MESSAGE), multiple_choice()],
57+
scorer=choice(),
58+
)
59+
60+
4161
def record_to_sample(record: dict[str, Any]) -> Sample:
4262
return Sample(
4363
input=record["Question"],

0 commit comments

Comments
 (0)