open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-29 17:35:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reduce values in NeedleHaystackCurriculum

Browse source
This commit is contained in:
Andreas Koepf 2025-03-16 15:52:48 +01:00
parent ed766028fb
commit 8a3c29e1ee
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/cognition/needle_haystack.py
View file

@ -137,7 +137,7 @@ class NeedleHaystackCurriculum(BaseCurriculum):
self._define_attributes(
RangeAttributeDefinition(
name="num_statements",
levels=[10, 100, 1_000, 10_000, 100_000, 1_000_000, 168_386_000],
levels=[10, 100, 500, 1_000, 5_000, 10_000, 50_000, 100_000],
description="Number of statements in the haystack",
lower_field_name="min_num_statements",
upper_field_name="max_num_statements",