Add groups to tests with errors.

evidentlyai · Jul 29, 2022 · aa05e40 · aa05e40
1 parent 9b8e565
commit aa05e40
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/evidently/suite/base_suite.py b/src/evidently/suite/base_suite.py
@@ -101,16 +101,15 @@ def run_checks(self):
 
         for test in self.context.execution_graph.get_test_execution_iterator():
             try:
-                test_result = test.check()
-                test_result.groups.update({
-                    GroupingTypes.TestGroup.id: test.group,
-                    GroupingTypes.TestType.id: test.name,
-                })
-                test_results[test] = test_result
+                test_results[test] = test.check()
             except BaseException as ex:
                 test_results[test] = TestResult(name=test.name,
                                                 status=TestResult.ERROR,
                                                 description=f"Test failed with exceptions: {ex}")
+            test_results[test].groups.update({
+                GroupingTypes.TestGroup.id: test.group,
+                GroupingTypes.TestType.id: test.name,
+            })
 
         self.context.test_results = test_results
         self.context.state = States.Tested