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

jugs jugs jugs lint

Browse source
This commit is contained in:
Rich Jones 2025-02-20 16:15:29 +01:00
parent 6f00690ae1
commit 0f798457ed
2 changed files with 32 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/algorithmic/jugs.py
View file

@ -232,9 +232,9 @@ class JugsConfig:
def validate(self):
"""Validate configuration parameters"""
assert self.num_jugs > 2, "edge_probability must be gt 2"
assert self.difficulty > 0, "edge_probability must be gt 0"
assert self.difficulty < 200, "edge_probability must be gt 200"
assert self.num_jugs > 2, "num_jugs must be gt 2"
assert self.difficulty > 0, "difficulty must be gt 0"
assert self.difficulty < 200, "difficulty must be lt 200"
class JugsDataset(ProceduralDataset):