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

pre-commit

Browse source
This commit is contained in:
Zafir Stojanovski 2025-02-21 13:39:05 +01:00
parent 941085e0c5
commit 96464388bb
3 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

18
reasoning_gym/algorithmic/binary_alternation.py
View file

@ -27,8 +27,8 @@ Now, determine the minimum number of swaps to make the following binary string a
class BinaryAlternationConfig:
"""Configuration for Count Bits dataset generation"""
min_n: int = 10 # Minimum number of bits in the binary string
max_n: int = 30 # Maximum number of bits in the binary string
min_n: int = 10 # Minimum number of bits in the binary string
max_n: int = 30 # Maximum number of bits in the binary string
p_solvable: float = 0.8 # Probability of generating a solvable sample
size: int = 500 # Virtual dataset size
@ -50,7 +50,7 @@ class BinaryAlternationDataset(ProceduralDataset):
def _get_binary_string(self, rng: Random, solvable: bool) -> str:
n = rng.randint(self.config.min_n, self.config.max_n)
ones, zeros = n // 2, n // 2
# Check if we need to add an extra bit
if n % 2 == 1:
if rng.random() < 0.5:
@ -69,13 +69,12 @@ class BinaryAlternationDataset(ProceduralDataset):
ones += 2
else:
zeros += 2
# Generate the string
string = ["1"] * ones + ["0"] * zeros
rng.shuffle(string)
return "".join(string)
def _get_answer(self, string: str) -> int:
"""Calculate the minimum number of swaps to make the string alternating"""
@ -85,11 +84,11 @@ class BinaryAlternationDataset(ProceduralDataset):
if c != expected:
incorrect += 1
expected = "1" if expected == "0" else "0"
return incorrect // 2 # number of swaps is half of incorrect positions
return incorrect // 2 # number of swaps is half of incorrect positions
ones, zeros = string.count("1"), string.count("0")
if abs(ones-zeros) > 1:
return -1 # impossible to make alternating
if abs(ones - zeros) > 1:
return -1 # impossible to make alternating
if ones > zeros:
return get_num_swaps("1")
elif ones < zeros:
@ -111,4 +110,5 @@ class BinaryAlternationDataset(ProceduralDataset):
"metadata": {"string": string, "solution": answer, "solvable": solvable},
}
register_dataset("binary_alternation", BinaryAlternationDataset, BinaryAlternationConfig)