Remove play file and format with pre-commit

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

View file

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

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'])

View file

@ -102,9 +102,12 @@ In solving the equations, please abide by the following instruction:
if len(real_solutions) > 0:
real_solutions.sort()
break
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 {
"question": question,

View file

@ -64,7 +64,7 @@ class PolynomialMultiplicationDataset(ProceduralDataset):
self.added_instruction = """
In addition, When doing calculation, Use the following instructions together with your mathematical ingenuity to solve the integral problems
## 1. Use ** instead ^ to represent powers. For example 7*X**2 instead of 7*X^2.
## 2. Always use * when doing all sorts of multiplcation in your reasoning steps and even in reporting answers.
## 2. Always use * when doing all sorts of multiplcation in your reasoning steps and even in reporting answers.
"""
super().__init__(config=config, seed=config.seed, size=config.size)