@@ -68,7 +68,6 @@ def init_codeflash() -> None:
6868 did_add_new_key = prompt_api_key ()
6969
7070 if should_modify_pyproject_toml ():
71-
7271 setup_info : SetupInfo = collect_setup_info ()
7372
7473 configure_pyproject_toml (setup_info )
@@ -81,7 +80,6 @@ def init_codeflash() -> None:
8180 if "setup_info" in locals ():
8281 module_string = f" you selected ({ setup_info .module_root } )"
8382
84-
8583 click .echo (
8684 f"{ LF } "
8785 f"⚡️ Codeflash is now set up! You can now run:{ LF } "
@@ -123,11 +121,13 @@ def ask_run_end_to_end_test(args: Namespace) -> None:
123121 bubble_sort_path , bubble_sort_test_path = create_bubble_sort_file_and_test (args )
124122 run_end_to_end_test (args , bubble_sort_path , bubble_sort_test_path )
125123
124+
126125def should_modify_pyproject_toml () -> bool :
127126 """Check if the current directory contains a valid pyproject.toml file with codeflash config
128127 If it does, ask the user if they want to re-configure it.
129128 """
130129 from rich .prompt import Confirm
130+
131131 pyproject_toml_path = Path .cwd () / "pyproject.toml"
132132 if not pyproject_toml_path .exists ():
133133 return True
@@ -142,7 +142,9 @@ def should_modify_pyproject_toml() -> bool:
142142 return True
143143
144144 create_toml = Confirm .ask (
145- "✅ A valid Codeflash config already exists in this project. Do you want to re-configure it?" , default = False , show_default = True
145+ "✅ A valid Codeflash config already exists in this project. Do you want to re-configure it?" ,
146+ default = False ,
147+ show_default = True ,
146148 )
147149 return create_toml
148150
0 commit comments