include ranges rather than sampled values in difficulty metadata dicts (#387)

* update difficulty metadata for logic datasets

* update difficulty metadata for graph datasets

* update difficulty metadata for geometry datasets

* update difficulty metadata for games datasets

* update difficulty metadata for cognition datasets

* update difficulty metadata for arithmetic datasets

* update difficulty metadata for arc datasets

* update difficulty metadata for algorithmic datasets

* update difficulty metadata for algebra datasets

* use tuples

* update tests

* update tests
This commit is contained in:
Oliver Stanley 2025-03-20 09:27:03 +00:00 committed by GitHub
parent b69c35818a
commit 7475a20700
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
80 changed files with 304 additions and 126 deletions

View file

@ -183,9 +183,11 @@ class StringManipulationDataset(ProceduralDataset):
"solution": answer,
"states": states,
"selected_rules": [rule for rule, _ in selected_rules],
"string_length": string_length,
"num_rules": num_rules,
"difficulty": {
"string_length": string_length,
"num_rules": num_rules,
"string_length": (self.config.min_string_length, self.config.max_string_length),
"num_rules": (self.config.min_num_rules, self.config.max_num_rules),
},
},
}