From 569b679adbae849c89fabaf04ddc9c598b395d27 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 1 Jul 2022 15:33:36 +0200 Subject: [PATCH] Update expected values in CodeGen tests (#17888) Co-authored-by: ydshieh --- tests/models/codegen/test_modeling_codegen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/codegen/test_modeling_codegen.py b/tests/models/codegen/test_modeling_codegen.py index 37919b0439..b59adc7818 100644 --- a/tests/models/codegen/test_modeling_codegen.py +++ b/tests/models/codegen/test_modeling_codegen.py @@ -503,7 +503,7 @@ class CodeGenModelLanguageGenerationTest(unittest.TestCase): output_seq_tt_strs = tokenizer.batch_decode(output_seq_tt, skip_special_tokens=True) if torch_device == "cuda": - EXPECTED_OUTPUT_STR = 'def hello_world():\n print("Hello World")\n\n#' + EXPECTED_OUTPUT_STR = 'def hello_world():\n print("Hello World")\n return True\n\nresult =' else: EXPECTED_OUTPUT_STR = "def hello_world():\r\n print('Hello, World.')\r\n\r\n\r"