error message for baseline establishment failure

Author: mohammedahmed18

codeflash/errors/errors.py

@@ -35,9 +35,11 @@ def function_optimization_attempted_error() -> CodeflashError:
     )
 
 
-def baseline_establishment_failed_error() -> CodeflashError:
+def baseline_establishment_failed_error(failure_msg: str) -> CodeflashError:
     return CodeflashError(
-        "BASELINE_ESTABLISHMENT_FAILED_ERROR", "Failed to establish a baseline for the original code."
+        "BASELINE_ESTABLISHMENT_FAILED_ERROR",
+        "Failed to establish a baseline for the original code. {failure_msg}",
+        **locals(),
     )
 
 

codeflash/optimization/function_optimizer.py

@@ -1461,6 +1461,7 @@ def establish_original_code_baseline(
                 )
             )
             console.rule()
+            failure_msg = ""
 
             total_timing = benchmarking_results.total_passed_runtime()  # caution: doesn't handle the loop index
             functions_to_remove = [
@@ -1469,17 +1470,17 @@ def establish_original_code_baseline(
                 if (result.test_type == TestType.GENERATED_REGRESSION and not result.did_pass)
             ]
             if total_timing == 0:
-                logger.warning(
-                    "The overall summed benchmark runtime of the original function is 0, couldn't run tests."
-                )
+                failure_msg = "The overall summed benchmark runtime of the original function is 0, couldn't run tests."
+                logger.warning(failure_msg)
                 console.rule()
                 success = False
             if not total_timing:
-                logger.warning("Failed to run the tests for the original function, skipping optimization")
+                failure_msg = "Failed to run the tests for the original function, skipping optimization"
+                logger.warning(failure_msg)
                 console.rule()
                 success = False
             if not success:
-                return Failure(baseline_establishment_failed_error())
+                return Failure(baseline_establishment_failed_error(failure_msg))
 
             loop_count = max([int(result.loop_index) for result in benchmarking_results.test_results])
             logger.info(