From 5f279be306d2de7b8ae25dc5c24d065f3f03ac4e Mon Sep 17 00:00:00 2001 From: abdulhakeem Date: Mon, 17 Feb 2025 03:13:06 -0600 Subject: [PATCH] Fix unit test --- tests/test_basic_arithmetic.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/test_basic_arithmetic.py b/tests/test_basic_arithmetic.py index 406e4617..757d3c2f 100644 --- a/tests/test_basic_arithmetic.py +++ b/tests/test_basic_arithmetic.py @@ -64,11 +64,7 @@ def test_arithmetic_dataset_format_styles(): max_digits=2, ) dataset = BasicArithmeticDataset(config) - assert all(item["question"].endswith("=") for item in dataset) - - config.format_style = "natural" - dataset = BasicArithmeticDataset(config) - assert all("=" in item["question"] for item in dataset) + assert all(item["question"].strip().endswith(".") for item in dataset) def test_arithmetic_dataset_iteration():