From 2dad5a56f31d320845686e43632cc445a4e4ed7a Mon Sep 17 00:00:00 2001 From: abdulhakeem Date: Thu, 13 Feb 2025 22:39:05 -0600 Subject: [PATCH] Remove play file and format with pre-commit --- .pre-commit-config.yaml | 2 +- play.py | 6 ------ reasoning_gym/algebra/polynomial_equations.py | 7 +++++-- reasoning_gym/algebra/polynomial_multiplication.py | 2 +- 4 files changed, 7 insertions(+), 10 deletions(-) delete mode 100644 play.py diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 49987591..b921ca29 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -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 diff --git a/play.py b/play.py deleted file mode 100644 index d7b42410..00000000 --- a/play.py +++ /dev/null @@ -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']) \ No newline at end of file diff --git a/reasoning_gym/algebra/polynomial_equations.py b/reasoning_gym/algebra/polynomial_equations.py index dad8a0b0..cd4842ee 100644 --- a/reasoning_gym/algebra/polynomial_equations.py +++ b/reasoning_gym/algebra/polynomial_equations.py @@ -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, diff --git a/reasoning_gym/algebra/polynomial_multiplication.py b/reasoning_gym/algebra/polynomial_multiplication.py index 67a914e8..6076c32a 100644 --- a/reasoning_gym/algebra/polynomial_multiplication.py +++ b/reasoning_gym/algebra/polynomial_multiplication.py @@ -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)