Merge branch 'main' into tracer-optimization

Author: KRRT7

codeflash/cli_cmds/cmd_init.py

@@ -45,6 +45,7 @@
     f"{LF}"
 )
 
+
 @dataclass(frozen=True)
 class SetupInfo:
     module_root: str
@@ -70,7 +71,6 @@ def init_codeflash() -> None:
         did_add_new_key = prompt_api_key()
 
         if should_modify_pyproject_toml():
-
             setup_info: SetupInfo = collect_setup_info()
 
             configure_pyproject_toml(setup_info)
@@ -83,7 +83,6 @@ def init_codeflash() -> None:
         if "setup_info" in locals():
             module_string = f" you selected ({setup_info.module_root})"
 
-
         click.echo(
             f"{LF}"
             f"⚡️ Codeflash is now set up! You can now run:{LF}"
@@ -125,11 +124,13 @@ def ask_run_end_to_end_test(args: Namespace) -> None:
         bubble_sort_path, bubble_sort_test_path = create_bubble_sort_file_and_test(args)
         run_end_to_end_test(args, bubble_sort_path, bubble_sort_test_path)
 
+
 def should_modify_pyproject_toml() -> bool:
     """Check if the current directory contains a valid pyproject.toml file with codeflash config
     If it does, ask the user if they want to re-configure it.
     """
     from rich.prompt import Confirm
+
     pyproject_toml_path = Path.cwd() / "pyproject.toml"
     if not pyproject_toml_path.exists():
         return True
@@ -144,7 +145,9 @@ def should_modify_pyproject_toml() -> bool:
         return True
 
     create_toml = Confirm.ask(
-        "✅ A valid Codeflash config already exists in this project. Do you want to re-configure it?", default=False, show_default=True
+        "✅ A valid Codeflash config already exists in this project. Do you want to re-configure it?",
+        default=False,
+        show_default=True,
     )
     return create_toml
 
@@ -160,7 +163,18 @@ def collect_setup_info() -> SetupInfo:
     # Check for the existence of pyproject.toml or setup.py
     project_name = check_for_toml_or_setup_file()
 
-    ignore_subdirs = ["venv", "node_modules", "dist", "build", "build_temp", "build_scripts", "env", "logs", "tmp", "__pycache__"]
+    ignore_subdirs = [
+        "venv",
+        "node_modules",
+        "dist",
+        "build",
+        "build_temp",
+        "build_scripts",
+        "env",
+        "logs",
+        "tmp",
+        "__pycache__",
+    ]
     valid_subdirs = [
         d for d in next(os.walk("."))[1] if not d.startswith(".") and not d.startswith("__") and d not in ignore_subdirs
     ]
@@ -262,13 +276,13 @@ def collect_setup_info() -> SetupInfo:
     benchmarks_options.append(create_benchmarks_option)
     benchmarks_options.append(custom_dir_option)
 
-
     benchmarks_answer = inquirer_wrapper(
         inquirer.list_input,
         message="Where are your performance benchmarks located? (benchmarks must be a sub directory of your tests root directory)",
         choices=benchmarks_options,
         default=(
-            default_benchmarks_subdir if default_benchmarks_subdir in benchmarks_options else benchmarks_options[0]),
+            default_benchmarks_subdir if default_benchmarks_subdir in benchmarks_options else benchmarks_options[0]
+        ),
     )
 
     if benchmarks_answer == create_benchmarks_option:
@@ -304,7 +318,6 @@ def collect_setup_info() -> SetupInfo:
     #         carousel=True,
     #     )
 
-
     formatter = inquirer_wrapper(
         inquirer.list_input,
         message="Which code formatter do you use?",
@@ -340,7 +353,7 @@ def collect_setup_info() -> SetupInfo:
     return SetupInfo(
         module_root=str(module_root),
         tests_root=str(tests_root),
-        benchmarks_root = str(benchmarks_root) if benchmarks_root else None,
+        benchmarks_root=str(benchmarks_root) if benchmarks_root else None,
         test_framework=cast(str, test_framework),
         ignore_paths=ignore_paths,
         formatter=cast(str, formatter),
@@ -499,19 +512,22 @@ def install_github_actions(override_formatter_check: bool = False) -> None:
             return
         workflows_path.mkdir(parents=True, exist_ok=True)
         from importlib.resources import files
+
         benchmark_mode = False
         if "benchmarks_root" in config:
             benchmark_mode = inquirer_wrapper(
                 inquirer.confirm,
                 message="⚡️It looks like you've configured a benchmarks_root in your config. Would you like to run the Github action in benchmark mode? "
-                        " This will show the impact of Codeflash's suggested optimizations on your benchmarks",
+                " This will show the impact of Codeflash's suggested optimizations on your benchmarks",
                 default=True,
             )
 
         optimize_yml_content = (
             files("codeflash").joinpath("cli_cmds", "workflows", "codeflash-optimize.yaml").read_text(encoding="utf-8")
         )
-        materialized_optimize_yml_content = customize_codeflash_yaml_content(optimize_yml_content, config, git_root, benchmark_mode)
+        materialized_optimize_yml_content = customize_codeflash_yaml_content(
+            optimize_yml_content, config, git_root, benchmark_mode
+        )
         with optimize_yaml_path.open("w", encoding="utf8") as optimize_yml_file:
             optimize_yml_file.write(materialized_optimize_yml_content)
         click.echo(f"{LF}✅ Created GitHub action workflow at {optimize_yaml_path}{LF}")
@@ -607,7 +623,7 @@ def get_dependency_manager_installation_string(dep_manager: DependencyManager) -
     python_version_string = f"'{py_version.major}.{py_version.minor}'"
     if dep_manager == DependencyManager.UV:
         return """name: 🐍 Setup UV
-        uses: astral-sh/setup-uv@v4
+        uses: astral-sh/setup-uv@v6
         with:
           enable-cache: true"""
     return f"""name: 🐍 Set up Python

codeflash/discovery/functions_to_optimize.py

@@ -22,6 +22,7 @@
     path_belongs_to_site_packages,
 )
 from codeflash.code_utils.git_utils import get_git_diff
+from codeflash.code_utils.time_utils import humanize_runtime
 from codeflash.discovery.discover_unit_tests import discover_unit_tests
 from codeflash.models.models import FunctionParent
 from codeflash.telemetry.posthog_cf import ph
@@ -203,6 +204,12 @@ def get_functions_to_optimize(
             functions, test_cfg.tests_root, ignore_paths, project_root, module_root, previous_checkpoint_functions
         )
         logger.info(f"Found {functions_count} function{'s' if functions_count > 1 else ''} to optimize")
+        if optimize_all:
+            three_min_in_ns = int(1.8e11)
+            logger.info(
+                f"It might take about {humanize_runtime(functions_count*three_min_in_ns)} to fully optimize this project. Codeflash "
+                f"will keep opening pull requests as it finds optimizations."
+            )
         return filtered_modified_functions, functions_count
 
 

docs/docs/getting-started/codeflash-github-actions.md

@@ -106,7 +106,7 @@ This assumes that you install poetry with pip and have Codeflash dependency in t
 2. uv
 
 ```yaml
-      - uses: astral-sh/setup-uv@v4
+      - uses: astral-sh/setup-uv@v6
         with:
           enable-cache: true
       - run: uv sync --group=dev