add metadata for caesar cipher, graph coloring, decimal arithmetic (#304)

* add metadata for caesar cipher, graph coloring, decimal arithmetic

* delete comma

* clean up variables
This commit is contained in:
vncntt 2025-03-09 10:08:56 -07:00 committed by GitHub
parent 1f360917fc
commit 91aa3f3ae2
3 changed files with 20 additions and 4 deletions

View file

@ -188,7 +188,12 @@ class DecimalArithmeticDataset(ProceduralDataset):
return {
"question": problem_str,
"answer": str(answer),
"metadata": {"decimal_places": decimal_places, "num_terms": terms},
"metadata": {
"difficulty": {
"decimal_places": decimal_places,
"num_terms": terms,
},
},
}
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float: