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

fix: Improve base conversion logic for non-standard bases

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-31 12:09:32 +01:00
parent e1be047d67
commit 7d911a8c25
2 changed files with 52 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

32
reasoning_gym/algorithmic/base_conversion.py
View file

@ -60,14 +60,34 @@ class BaseConversionDataset(ProceduralDataset):
value, source_base, target_base = self._generate_conversion(rng)
# Convert decimal to source base representation
source_repr = format(value, f"x" if source_base == 16 else f"b" if source_base == 2 else "").strip()
if source_base not in (2, 16):
source_repr = format(value, f"{source_base}x").lower().strip()
if source_base == 16:
source_repr = format(value, 'x')
elif source_base == 2:
source_repr = format(value, 'b')
else:
# Manual conversion for other bases
n = value
digits = []
while n:
digits.append(int(n % source_base))
n //= source_base
source_repr = ''.join(str(d) if d < 10 else chr(ord('a') + d - 10)
for d in reversed(digits) or [0])
# Convert decimal to target base for answer
target_repr = format(value, f"x" if target_base == 16 else f"b" if target_base == 2 else "").strip()
if target_base not in (2, 16):
target_repr = format(value, f"{target_base}x").lower().strip()
if target_base == 16:
target_repr = format(value, 'x')
elif target_base == 2:
target_repr = format(value, 'b')
else:
# Manual conversion for other bases
n = value
digits = []
while n:
digits.append(int(n % target_base))
n //= target_base
target_repr = ''.join(str(d) if d < 10 else chr(ord('a') + d - 10)
for d in reversed(digits) or [0])
source_name = self._format_base_name(source_base)
target_name = self._format_base_name(target_base)