Merge pull request codeflash-ai#519 from codeflash-ai/prevent-yet-another-crash

misrasaurabh1 · web-flow · commit 47d3ff154c9e · 2025-07-06T15:42:53.000-07:00
prevent another crash
diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py
@@ -312,6 +312,8 @@ def run(self) -> None:
                             original_module_ast=original_module_ast,
                             original_module_path=original_module_path,
                         )
+                        if function_optimizer is None:
+                            continue
 
                         self.current_function_optimizer = (
                             function_optimizer  # needed to clean up from the outside of this function

Original file line number	Diff line number	Diff line change
`@@ -312,6 +312,8 @@ def run(self) -> None:`
`312`	`312`	`original_module_ast=original_module_ast,`
`313`	`313`	`original_module_path=original_module_path,`
`314`	`314`	`)`
	`315`	`+ if function_optimizer is None:`
	`316`	`+ continue`
`315`	`317`
`316`	`318`	`self.current_function_optimizer = (`
`317`	`319`	`function_optimizer # needed to clean up from the outside of this function`