mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
small change to word sequence reversal prompt (#252)
corrected ansewr format
This commit is contained in:
parent
3149edf2c4
commit
49db4ed761
1 changed files with 3 additions and 1 deletions
|
|
@ -8,6 +8,8 @@ from typing import Optional
|
|||
from ..data import read_data_file
|
||||
from ..factory import ProceduralDataset, register_dataset
|
||||
|
||||
QUESTION_TEMPLATE = """Solve the following problem. Provide you answer as a comma-separated list of word with a space the comma. Reverse this list of words: {question}"""
|
||||
|
||||
|
||||
@dataclass
|
||||
class WordSequenceReversalConfig:
|
||||
|
|
@ -49,7 +51,7 @@ class WordSequenceReversalDataset(ProceduralDataset):
|
|||
answer = ", ".join(reversed(words))
|
||||
|
||||
return {
|
||||
"question": f"Reverse this list of words: {question}",
|
||||
"question": f"{QUESTION_TEMPLATE.format(question=question)}",
|
||||
"answer": answer,
|
||||
"metadata": {"num_words": num_words, "words": words},
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue