open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-22 16:49:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor code formatting for improved readability across multiple files

Browse source
This commit is contained in:
Ritvik19 2026-04-18 16:50:18 +05:30
parent 6eb252ae32
commit c63a596489
16 changed files with 50 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

7
reasoning_gym/combinatorics/combinatorics.py
View file

@ -16,9 +16,7 @@ class CombinatoricsConfig:
min_n: int = 5
max_n: int = 15
task_types: tuple[str, ...] = TASK_TYPES
task_weights: list[float] = field(
default_factory=lambda: [0.2, 0.15, 0.2, 0.2, 0.15, 0.1]
)
task_weights: list[float] = field(default_factory=lambda: [0.2, 0.15, 0.2, 0.2, 0.15, 0.1])
seed: Optional[int] = None
size: int = 500
@ -40,8 +38,7 @@ class CombinatoricsDataset(ProceduralDataset):
k = rng.randint(1, n - 1)
answer = math.comb(n, k)
question = (
f"How many ways can you choose {k} items from a set of {n} items? "
f"Give your answer as a single integer."
f"How many ways can you choose {k} items from a set of {n} items? " f"Give your answer as a single integer."
)
return {"question": question, "answer": str(answer), "task_type": "ncr"}