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

Add validation

Browse source
This commit is contained in:
Oliver 2025-02-23 22:23:45 +00:00
parent 342902683f
commit e07287e1f9
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/code/codeio.py
View file

@ -60,7 +60,7 @@ class CodeIOConfig:
def validate(self) -> None:
"""Validate configuration parameters"""
pass
assert 0.0 <= self.input_prediction_probability <= 1.0, "input_prediction_probability must be in [0, 1]"
class CodeIODataset(ProceduralDataset):
@ -146,4 +146,4 @@ class CodeIODataset(ProceduralDataset):
# Register the dataset
register_dataset("codeio", CodeIODataset, CodeIOConfig)
# register_dataset("codeio", CodeIODataset, CodeIOConfig)