diff --git a/codeflash/cli_cmds/cmd_init.py b/codeflash/cli_cmds/cmd_init.py index 2a9b977e2..63546c0d1 100644 --- a/codeflash/cli_cmds/cmd_init.py +++ b/codeflash/cli_cmds/cmd_init.py @@ -587,6 +587,11 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None: ) elif formatter == "don't use a formatter": formatter_cmds.append("disabled") + if formatter in ["black", "ruff"]: + try: + result = subprocess.run([formatter], capture_output=True, check=False) + except FileNotFoundError as e: + click.echo(f"⚠️ Formatter not found: {formatter}, please ensure it is installed") codeflash_section["formatter-cmds"] = formatter_cmds # Add the 'codeflash' section, ensuring 'tool' section exists tool_section = pyproject_data.get("tool", tomlkit.table())