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 parameters to _generate_sentence_dataset

Browse source
This commit is contained in:
abdulhakeem 2025-01-25 23:17:39 -06:00
parent 384a00ec71
commit 4d50cfd514
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/algorithmic/sentence_reordering.py
View file

@ -42,6 +42,8 @@ class SentenceReorderingDataset(ProceduralDataset):
Args:
sentence (str): The correct sentence to use for dataset generation.
seed (int): The seed to use for random number generation.
idx (int): The index to add to the seed for random number generation.
shuffle (bool): Whether to shuffle the words to create the input sentence.
Returns: