From 5c45e5534052b8c2d4d2c755da5bc9b9d14e3538 Mon Sep 17 00:00:00 2001 From: AhmedSaif2 Date: Fri, 21 Feb 2025 15:50:03 +0200 Subject: [PATCH] extend format tests to allow questions that ends with question marks --- tests/test_basic_arithmetic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_basic_arithmetic.py b/tests/test_basic_arithmetic.py index c1035af9..6eda876a 100644 --- a/tests/test_basic_arithmetic.py +++ b/tests/test_basic_arithmetic.py @@ -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():