Merge pull request #859 from codeflash-ai/line-profiler-timeout-handling

Return default line profiler output when timeout happens

Author: aseembits93

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 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": ""}
         if line_profile_results["str_out"] == "":
             logger.warning(
                 f"Couldn't run line profiler for original function {self.function_to_optimize.function_name}"