da rulez

Author: KRRT7

codeflash-benchmark/codeflash_benchmark/__init__.py

@@ -1,3 +1,3 @@
 """CodeFlash Benchmark - Pytest benchmarking plugin for codeflash.ai."""
 
-__version__ = "0.1.0"
+__version__ = "0.1.0"

codeflash-benchmark/codeflash_benchmark/plugin.py

@@ -25,7 +25,6 @@ def pytest_addoption(parser: pytest.Parser) -> None:
     )
 
 
-
 @pytest.fixture
 def benchmark(request: pytest.FixtureRequest) -> object:
     """Benchmark fixture that works with or without pytest-benchmark installed."""

codeflash/benchmarking/plugin/plugin.py

@@ -285,4 +285,4 @@ def _run_benchmark(self, func, *args, **kwargs):  # noqa: ANN001, ANN002, ANN003
             return result
 
 
-codeflash_benchmark_plugin = CodeFlashBenchmarkPlugin()
+codeflash_benchmark_plugin = CodeFlashBenchmarkPlugin()

codeflash/discovery/functions_to_optimize.py

@@ -204,6 +204,7 @@ def get_functions_to_optimize(
                 functions[file] = [found_function]
         else:
             logger.info("Finding all functions modified in the current git diff ...")
+            console.rule()
             ph("cli-optimizing-git-diff")
             functions = get_functions_within_git_diff()
         filtered_modified_functions, functions_count = filter_functions(

codeflash/optimization/function_optimizer.py

@@ -840,6 +840,7 @@ def instrument_existing_tests(self, function_to_all_tests: dict[str, set[Functio
                 f"{concolic_coverage_test_files_count} concolic coverage test file"
                 f"{'s' if concolic_coverage_test_files_count != 1 else ''} for {func_qualname}"
             )
+            console.rule()
         return unique_instrumented_test_files
 
     def generate_tests_and_optimizations(

codeflash/optimization/optimizer.py

@@ -106,7 +106,7 @@ def run_benchmarks(
                 for file in file_path_to_source_code:
                     with file.open("w", encoding="utf8") as f:
                         f.write(file_path_to_source_code[file])
-
+        console.rule()
         return function_benchmark_timings, total_benchmark_timings
 
     def get_optimizable_functions(self) -> tuple[dict[Path, list[FunctionToOptimize]], int, Path | None]: