mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
extend format tests to allow questions that ends with question marks
This commit is contained in:
parent
73dfaa7f58
commit
5c45e55340
1 changed files with 1 additions and 1 deletions
|
|
@ -74,7 +74,7 @@ def test_arithmetic_dataset_format_styles():
|
||||||
max_digits=2,
|
max_digits=2,
|
||||||
)
|
)
|
||||||
dataset = BasicArithmeticDataset(config)
|
dataset = BasicArithmeticDataset(config)
|
||||||
assert all(item["question"].strip().endswith(".") for item in dataset)
|
assert all(item["question"].strip().endswith(".") or item["question"].strip().endswith("?") for item in dataset)
|
||||||
|
|
||||||
|
|
||||||
def test_arithmetic_dataset_iteration():
|
def test_arithmetic_dataset_iteration():
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue