Skip to content

Commit 18e393b

Browse files
authored
Avoid eval mode if not present (#33)
1 parent 76b5e28 commit 18e393b

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

generate_zeroshot_evals.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -105,16 +105,18 @@ async def evaluate(
105105
)
106106
)
107107

108-
results.append(
109-
{
110-
"uuid": query.id,
111-
"question": query.question,
112-
"predicted": query.predicted,
113-
"target": query.target,
114-
"unsure": query.unsure,
115-
"evaluation_mode": query.evaluation_mode,
116-
}
117-
)
108+
result_dict = {
109+
"uuid": query.id,
110+
"question": query.question,
111+
"predicted": query.predicted,
112+
"target": query.target,
113+
"unsure": query.unsure,
114+
}
115+
116+
if query.evaluation_mode is not None:
117+
result_dict["evaluation_mode"] = query.evaluation_mode
118+
119+
results.append(result_dict)
118120

119121
# make directory if it doesn't exist
120122
if not os.path.exists(output_dir):

0 commit comments

Comments
 (0)