From 92b9dee6d225d26a3c1ae63867d928436400fea9 Mon Sep 17 00:00:00 2001 From: Jakub Kosek Date: Fri, 1 Sep 2023 07:23:49 -0700 Subject: [PATCH] Improved L1 test for NeMo example --- .../L1_example_nemo_megatron_gpt_prompt_learning/test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/functional/L1_example_nemo_megatron_gpt_prompt_learning/test.py b/tests/functional/L1_example_nemo_megatron_gpt_prompt_learning/test.py index f03f3f0..4798870 100755 --- a/tests/functional/L1_example_nemo_megatron_gpt_prompt_learning/test.py +++ b/tests/functional/L1_example_nemo_megatron_gpt_prompt_learning/test.py @@ -42,7 +42,7 @@ def verify_client_output(client_output): else: LOGGER.info(f'Found "{expected_pattern}" in client output') - expected_patterns = [r"positive", r"alarm_set", r"time\(seven am\)"] + expected_patterns = [r"neutral", r"set the alarm", r"seven am"] for expected_pattern in expected_patterns: output_match = re.search(expected_pattern, client_output, re.MULTILINE) output_array = output_match.group(0) if output_match else None