precommit fix

Author: aseembits93

codeflash/cli_cmds/cmd_init.py

@@ -721,7 +721,7 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
         )
     elif formatter == "don't use a formatter":
         formatter_cmds.append("disabled")
-    check_formatter_installed(formatter_cmds)
+    check_formatter_installed(formatter_cmds, exit_on_failure="no")
     codeflash_section["formatter-cmds"] = formatter_cmds
     # Add the 'codeflash' section, ensuring 'tool' section exists
     tool_section = pyproject_data.get("tool", tomlkit.table())

codeflash/code_utils/env_utils.py

@@ -12,7 +12,7 @@
 from codeflash.code_utils.shell_utils import read_api_key_from_shell_config
 
 
-def check_formatter_installed(formatter_cmds: list[str]) -> bool:
+def check_formatter_installed(formatter_cmds: list[str], exit_on_failure: str = "yes") -> bool:
     return_code = True
     if formatter_cmds[0] == "disabled":
         return return_code
@@ -27,7 +27,8 @@ def check_formatter_installed(formatter_cmds: list[str]) -> bool:
             print(
                 "⚠️ Codeflash requires a code formatter to be installed in your environment, but none was found. Please install a supported formatter, verify the formatter-cmds in your codeflash pyproject.toml config and try again."
             )
-            sys.exit(1)
+            if exit_on_failure == "yes":
+                sys.exit(1)
     return return_code
 
 

codeflash/code_utils/formatter.py

@@ -31,7 +31,7 @@ def format_code(formatter_cmds: list[str], path: Path) -> str:
                 console.rule(f"Formatted Successfully with: {formatter_name.replace('$file', path.name)}")
             else:
                 logger.error(f"Failed to format code with {' '.join(formatter_cmd_list)}")
-        except (FileNotFoundError, NotADirectoryError) as e:
+        except FileNotFoundError as e:
             from rich.panel import Panel
             from rich.text import Text