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

Apply black formatting fixes

Browse source
This commit is contained in:
Andreas Koepf 2026-03-15 13:49:25 +01:00
parent 6a9f3860a0
commit 6e030c174a
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/algorithmic/cryptarithm.py
View file

@ -95,7 +95,10 @@ def verify_cryptarithm_solution(
# Check if sum is correct
computed_sum = sum(word_numbers)
if computed_sum != result_number:
return False, f"Arithmetic equation not satisfied: {word_numbers} sums to {computed_sum}, expected {result_number}"
return (
False,
f"Arithmetic equation not satisfied: {word_numbers} sums to {computed_sum}, expected {result_number}",
)
except (KeyError, ValueError) as e:
return False, f"Error applying mapping: {e}"

4
tests/test_cryptarithm.py
View file

@ -125,7 +125,9 @@ def test_cryptarithm_score_answer():
for pair in correct_answer_str.split(","):
alpha, num_str = pair.split("=")
correct_mapping[alpha] = int(num_str)
reversed_answer = ",".join(f"{letter}={correct_mapping[letter]}" for letter in reversed(sorted(correct_mapping.keys())))
reversed_answer = ",".join(
f"{letter}={correct_mapping[letter]}" for letter in reversed(sorted(correct_mapping.keys()))
)
score = dataset.score_answer(answer=reversed_answer, entry=puzzle)
assert score == 1.0, f"Expected 1.0 for correct answer in different order, got {score}"