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

Remove play file and format with pre-commit

Browse source
This commit is contained in:
abdulhakeem 2025-02-13 22:39:05 -06:00
parent 0738d3fd17
commit 2dad5a56f3
4 changed files with 7 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.pre-commit-config.yaml
View file

@ -11,7 +11,7 @@ repos:
rev: 24.1.1 rev: 24.1.1
hooks: hooks:
- id: black - id: black
language_version: python3.11 language_version: python3.13
- repo: https://github.com/pycqa/isort - repo: https://github.com/pycqa/isort
rev: 5.13.2 rev: 5.13.2

6
play.py
View file

@ -1,6 +0,0 @@
import reasoning_gym
data = reasoning_gym.create_dataset('polynomial_equations', size=3, seed=42)
for i, x in enumerate(data):
print(f"{i}: question={x['question']}\n")
print(f"{i}: answer={x['answer']}\n")
print('metadata:', x['metadata'])

5
reasoning_gym/algebra/polynomial_equations.py
View file

@ -104,7 +104,10 @@ In solving the equations, please abide by the following instruction:
break break
answer_str = ", ".join(str(x) for x in real_solutions) answer_str = ", ".join(str(x) for x in real_solutions)
question = rng.choice(self._prompt_templates).format(variable=variable, polynomial_expanded=polynomial_expanded) + self.added_instruction question = (
rng.choice(self._prompt_templates).format(variable=variable, polynomial_expanded=polynomial_expanded)
+ self.added_instruction
)
return { return {
"question": question, "question": question,