From 4a88798d2322feaee9dc923f3c33ea619355bb1e Mon Sep 17 00:00:00 2001 From: Aseem Saxena Date: Tue, 28 Oct 2025 11:17:24 -0700 Subject: [PATCH 1/2] return default line profiler output when timeout happens --- codeflash/optimization/function_optimizer.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index 1825798bd..734efc035 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -2044,6 +2044,13 @@ def line_profiler_step( self.write_code_and_helpers( self.function_to_optimize_source_code, original_helper_code, self.function_to_optimize.file_path ) + # this will happen when a timeoutexpired exception happens + if isinstance(line_profile_results, TestResults) and not line_profile_results.test_results: + logger.warning( + f"Timeout occured while running line profiler for original function {self.function_to_optimize.function_name}" + ) + # set default value for line profiler results + return {"timings": {}, "unit": 0, "str_out": ""} if line_profile_results["str_out"] == "": logger.warning( f"Couldn't run line profiler for original function {self.function_to_optimize.function_name}" From 423f80a678ddb7777bcf25fe75e867a63bfda739 Mon Sep 17 00:00:00 2001 From: Aseem Saxena Date: Tue, 28 Oct 2025 11:21:47 -0700 Subject: [PATCH 2/2] Apply suggestion from @aseembits93 --- codeflash/optimization/function_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index 734efc035..0ac3ba44a 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -2047,7 +2047,7 @@ def line_profiler_step( # this will happen when a timeoutexpired exception happens if isinstance(line_profile_results, TestResults) and not line_profile_results.test_results: logger.warning( - f"Timeout occured while running line profiler for original function {self.function_to_optimize.function_name}" + f"Timeout occurred while running line profiler for original function {self.function_to_optimize.function_name}" ) # set default value for line profiler results return {"timings": {}, "unit": 0, "str_out": ""}