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

add metadata for caesar cipher, graph coloring, decimal arithmetic (#304)

Browse source 
* add metadata for caesar cipher, graph coloring, decimal arithmetic

* delete comma

* clean up variables
This commit is contained in:
vncntt 2025-03-09 10:08:56 -07:00 • committed by GitHub
parent fc908d4cf4
commit af6120c095
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/algorithmic/caesar_cipher.py
View file

@ -67,6 +67,7 @@ class CaesarCipherDataset(ProceduralDataset):
# Select random sentence and rotation
sentence = rng.choice(self.valid_sentences)
num_words = len(sentence.split())
rotation = rng.randint(self.config.min_rotation, self.config.max_rotation)
# Generate cipher text
@ -79,6 +80,10 @@ class CaesarCipherDataset(ProceduralDataset):
"rotation": rotation,
"cipher_text": cipher_text,
"clear_text": sentence,
"difficulty": {
"rotation": rotation,
"words": num_words,
},
},
}