open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-29 17:35:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added updates

Browse source
This commit is contained in:
joesharratt1229 2025-03-29 08:07:57 +00:00
parent 774d23664d
commit 9f9f816902
10 changed files with 461 additions and 503 deletions
Download patch file Download diff file Expand all files Collapse all files

4
training/utils/datasets.py
View file

@ -79,7 +79,7 @@ class ReasoningGymDataset(Dataset):
raise ValueError("Cannot update difficulty: dataset is not a CurriculumExperiment")
if method not in ["increment", "decrement"]:
raise ValueError("Invalid method: must be 'increment' or 'decrement'")
self.experiment.score_board.clear()
self.experiment.score_board.clear(dataset_name)
self.experiment.update_difficulty(dataset_name, method)
self.data = self.experiment.composite
return True
@ -98,7 +98,6 @@ class ReasoningGymDataset(Dataset):
first_key = list(scores.keys())[0]
output_results[key]["results"] = np.mean(scores[first_key])
output_results[key]["total_samples"] = value.total_scores
return output_results
@ -119,5 +118,4 @@ def make_dataset(
kwargs["experiment"] = data_source
else:
kwargs["procedural_dataset"] = data_source
print(type(data_source))
return ReasoningGymDataset(**kwargs)