open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add difficulty metadata

Browse source
This commit is contained in:
Zafir Stojanovski 2025-03-14 22:22:30 +01:00 • committed by Rich Jones
parent 29bf78293f
commit 814da6e08a
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/cognition/figlet_fonts.py
View file

@ -185,7 +185,11 @@ class FigletFontDataset(ProceduralDataset):
return {
"question": rng.choice(self._prompt_templates).format(figlet_render=figlet_render),
"answer": word,
"metadata": {"font": chosen_font, "space_letters": self.config.space_letters},
"metadata": {
"font": chosen_font,
"space_letters": self.config.space_letters,
"difficulty": {"word_len": len(word)},
},
}
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float: