mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-23 16:55:05 +00:00
fix: Add validate() method to QuantumLockConfig
This commit is contained in:
parent
970724fc49
commit
38d64649f5
1 changed files with 5 additions and 0 deletions
|
|
@ -15,6 +15,11 @@ class QuantumLockConfig:
|
|||
seed: Optional[int] = None
|
||||
size: int = 500
|
||||
|
||||
def validate(self) -> None:
|
||||
"""Validate configuration parameters"""
|
||||
assert self.difficulty > 0, "difficulty must be positive"
|
||||
assert self.size > 0, "size must be positive"
|
||||
|
||||
|
||||
class QuantumLockDataset(ProceduralDataset):
|
||||
"""Generates QuantumLock tasks"""
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue