From 1f6bb04fc8b0c994e223b23938a979c6e9c5d0b7 Mon Sep 17 00:00:00 2001 From: Sukriti-Sharma4 Date: Thu, 23 May 2024 22:02:55 -0600 Subject: [PATCH] fix tests Signed-off-by: Sukriti-Sharma4 --- tests/utils/test_data_utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/utils/test_data_utils.py b/tests/utils/test_data_utils.py index f20790895..3857fd533 100644 --- a/tests/utils/test_data_utils.py +++ b/tests/utils/test_data_utils.py @@ -31,7 +31,8 @@ def test_formatting_function(): template = "### Input: {{Tweet text}} \n\n ### Response: {{text_label}}" # First response from the data file that is read. expected_response = ( - "### Input: No this is my first job \n\n ### Response: no complaint" + "### Input: @HMRCcustomers No this is my first job" + + " \n\n ### Response: no complaint" ) formatted_dataset, dataset_text_field = data_utils.formatting_function( json_dataset, template @@ -46,7 +47,8 @@ def test_formatting_function_adds_eos_token(): template = "### Input: {{Tweet text}} \n\n ### Response: {{text_label}}" # First response from the data file that is read. expected_response = ( - "### Input: No this is my first job \n\n ### Response: no complaintEOS" + "### Input: @HMRCcustomers No this is my first job" + + " \n\n ### Response: no complaintEOS" ) formatted_dataset, dataset_text_field = data_utils.formatting_function( json_dataset, template, "EOS"