Fix regex (#1178)

mlcommons · Jul 18, 2022 · 67eb6a0 · 67eb6a0
1 parent 5dd9ed7
commit 67eb6a0
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/compliance/nvidia/TEST01/verify_performance.py b/compliance/nvidia/TEST01/verify_performance.py
@@ -51,12 +51,12 @@ def main():
             continue
 
         if ref_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 continue
 
         if ref_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 continue
 
@@ -89,12 +89,12 @@ def main():
             continue
 
         if test_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 continue
 
         if test_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 continue
 

diff --git a/compliance/nvidia/TEST04/verify_performance.py b/compliance/nvidia/TEST04/verify_performance.py
@@ -50,13 +50,13 @@ def main():
             continue
 
         if ref_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 ref_score = 1e9 / float(ref_score)
                 continue
 
         if ref_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 ref_score = 1e9 / float(ref_score)
                 continue
@@ -90,13 +90,13 @@ def main():
             continue
 
         if test_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 test_score = 1e9 / float(test_score)
                 continue
 
         if test_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 test_score = 1e9 / float(test_score)
                 continue

diff --git a/compliance/nvidia/TEST05/verify_performance.py b/compliance/nvidia/TEST05/verify_performance.py
@@ -51,12 +51,12 @@ def main():
             continue
 
         if ref_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 continue
 
         if ref_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 ref_score = line.split(": ",1)[1].strip()
                 continue
 
@@ -89,12 +89,12 @@ def main():
             continue
 
         if test_mode == "SingleStream":
-            if re.match("Early stopping 90th percentile estimate", line):
+            if re.match(".*Early stopping 90th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 continue
 
         if test_mode == "MultiStream":
-            if re.match("Early stopping 99th percentile estimate", line):
+            if re.match(".*Early stopping 99th percentile estimate", line):
                 test_score = line.split(": ",1)[1].strip()
                 continue