extend format tests to allow questions that ends with question marks

This commit is contained in:
AhmedSaif2 2025-02-21 15:50:03 +02:00
parent 73dfaa7f58
commit 5c45e55340

View file

@ -74,7 +74,7 @@ def test_arithmetic_dataset_format_styles():
max_digits=2,
)
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():