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

CeasarCipher text indexing

Browse source
This commit is contained in:
EduardDurech 2025-02-09 08:32:14 +00:00
parent 5279ccf7e1
commit ca0fb97884
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/curricula/algorithmic/caesar_cipher_curriculum.py
View file

@ -105,8 +105,9 @@ class CaesarCipherCurriculum(BaseCurriculum):
)
)[-1],
"read_text": lambda refs: (
idx := refs["dataset_rng"].randint(0, len(self.text_data) - 1),
" ".join(self.text_data[idx:idx+refs["num_words"]()])
n_words := refs["num_words"](),
idx := refs["dataset_rng"].randint(0, len(self.text_data) - n_words),
" ".join(self.text_data[idx:idx+n_words])
)[-1]
}
}