mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
Basic curriculum (#198)
* feat: Add optional curriculum support to dataset registration and creation * docs: Add docstrings to create_curriculum() and register_dataset() * feat: Add curriculum configuration classes for CurriculumExperiment * feat: Add weight parameter to CurriculumAttributeConfig and use in DatasetSpec * refactor: Simplify CurriculumAttributeConfig with "*" attribute level support * test: Add unit tests for CurriculumExperiment class * feat: Add from_yaml() method to CurriculumExperimentConfig with unit test
This commit is contained in:
parent
cbfdf097a0
commit
c69bc5d4e6
29 changed files with 943 additions and 63 deletions
|
|
@ -35,7 +35,7 @@ def test_experiment_management():
|
|||
exp = registry.get_experiment("test_exp")
|
||||
assert exp is not None
|
||||
assert exp.name == "test_exp"
|
||||
assert isinstance(exp.dataset, CompositeDataset)
|
||||
assert isinstance(exp.composite, CompositeDataset)
|
||||
assert exp.config == config
|
||||
|
||||
# Test removal
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue