mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-22 16:49:06 +00:00
fix: Add validation for size parameter in ABConfig
This commit is contained in:
parent
06e2745ceb
commit
c6ef97ec0b
2 changed files with 15 additions and 12 deletions
|
|
@ -58,7 +58,8 @@ class ABConfig:
|
|||
|
||||
def validate(self) -> None:
|
||||
"""Validate configuration parameters"""
|
||||
assert self.length > 0, "difficulty must be greater than 0"
|
||||
assert self.length > 0, "length must be greater than 0"
|
||||
assert self.size > 0, "size must be greater than 0"
|
||||
|
||||
|
||||
class ABDataset(ProceduralDataset):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue