Improved L1 test for NeMo example

triton-inference-server · Sep 1, 2023 · 92b9dee · 92b9dee
1 parent 0028fdf
commit 92b9dee
Showing 1 changed file with 1 addition and 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
@@ -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