PyTorch: Also capture tests failing with signal

easybuilders · Aug 4, 2022 · 706d230 · 706d230
1 parent 1989c86
commit 706d230
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -264,11 +264,9 @@ def test_step(self):
         (tests_out, tests_ec) = super(EB_PyTorch, self).test_step(return_output_ec=True)
 
         ran_tests_hits = re.findall(r"^Ran (?P<test_cnt>[0-9]+) tests in", tests_out, re.M)
-        test_cnt = 0
-        for hit in ran_tests_hits:
-            test_cnt += int(hit)
+        test_cnt = sum(int(hit) for hit in ran_tests_hits)
 
-        failed_tests = nub(re.findall(r"^(?P<failed_test_name>.*) failed!(?: Received signal: \w+)?\s*$", tests_out, re.M))
+        failed_tests = nub(re.findall(r"^(?P<test_name>.*) failed!(?: Received signal: \w+)?\s*$", tests_out, re.M))
         failed_test_cnt = len(failed_tests)
 
         if failed_test_cnt: