Update definitions.py

definitions.py

@@ -550,8 +550,10 @@ if __name__ == "__main__":
                 return {
                     "test_cases": test_generation["result"],
                     "test_results": test_results["result"],
+                }
             except Exception as e:
                 raise Exception(f"Testing stage failed: {str(e)}")
+                
 
         def _validate_stage_output(self, stage: PipelineStage, result: Dict[str, Any]):
             """Validate stage output against defined criteria"""