Update codeflash/code_utils/edit_generated_tests.py

Author: misrasaurabh1

codeflash/code_utils/edit_generated_tests.py

@@ -13,6 +13,7 @@
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.time_utils import format_perf, format_time
 from codeflash.models.models import GeneratedTests, GeneratedTestsList
+from codeflash.result.critic import performance_gain
 
 if TYPE_CHECKING:
     from codeflash.models.models import InvocationId