improve

PaddlePaddle · Jul 13, 2023 · aad2275 · aad2275
1 parent 4f14064
commit aad2275
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/extract_errors.py b/tests/extract_errors.py
@@ -4,11 +4,12 @@
 error_msg = sys.stdin.read()
 
 pattern = r'File "?(.*?)"?, line (\d+),.*\n(.*?)\n(.*?)$'
-match = re.search(pattern, error_msg, re.MULTILINE)
-if match:
+for match in re.finditer(pattern, error_msg, re.MULTILINE):
     file = match.group(1)
+    if not file.startswith("/"):
+        file = f"tests/{file}"
     line = match.group(2)
     error_info = match.group(4)
     # error_info = match.group(3) + '\n' + match.group(4)
-    output = f"::error file=tests/{file},line={line}::{error_info}"
+    output = f"::error file={file},line={line}::{error_info}"
     print(output)