remove debug prints

Author: KRRT7

codeflash/benchmarking/plugin/plugin.py

@@ -285,20 +285,4 @@ def _run_benchmark(self, func, *args, **kwargs):  # noqa: ANN001, ANN002, ANN003
             return result
 
 
-codeflash_benchmark_plugin = CodeFlashBenchmarkPlugin()
-
-
-# def pytest_configure(config: pytest.Config) -> None:
-#     """Register the benchmark marker and disable conflicting plugins."""
-#     config.addinivalue_line("markers", "benchmark: mark test as a benchmark that should be run with codeflash tracing")
-
-#     if config.getoption("--codeflash-trace") and PYTEST_BENCHMARK_INSTALLED:
-#         config.option.benchmark_disable = True
-#         config.pluginmanager.set_blocked("pytest_benchmark")
-#         config.pluginmanager.set_blocked("pytest-benchmark")
-
-
-# def pytest_addoption(parser: pytest.Parser) -> None:
-#     parser.addoption(
-#         "--codeflash-trace", action="store_true", default=False, help="Enable CodeFlash tracing for benchmarks"
-#     )
+codeflash_benchmark_plugin = CodeFlashBenchmarkPlugin()

codeflash/benchmarking/trace_benchmarks.py

@@ -12,7 +12,6 @@
 def trace_benchmarks_pytest(
     benchmarks_root: Path, tests_root: Path, project_root: Path, trace_file: Path, timeout: int = 300
 ) -> None:
-    logger.info("Tracing benchmarks with codeflash-benchmark")
     benchmark_env = os.environ.copy()
     if "PYTHONPATH" not in benchmark_env:
         benchmark_env["PYTHONPATH"] = str(project_root)
@@ -33,8 +32,6 @@ def trace_benchmarks_pytest(
         env=benchmark_env,
         timeout=timeout,
     )
-    logger.info(f"ran with args: {result.args}")
-    logger.info(f"Pytest output: {result.stderr}")
     if result.returncode != 0:
         if "ERROR collecting" in result.stdout:
             # Pattern matches "===== ERRORS =====" (any number of =) and captures everything after

codeflash/optimization/optimizer.py

@@ -74,7 +74,6 @@ def run_benchmarks(
             for file in file_to_funcs_to_optimize:
                 with file.open("r", encoding="utf8") as f:
                     file_path_to_source_code[file] = f.read()
-            console.print(f"Instrumented {len(file_to_funcs_to_optimize)} files for benchmarking…")
             try:
                 instrument_codeflash_trace_decorator(file_to_funcs_to_optimize)
                 trace_file = Path(self.args.benchmarks_root) / "benchmarks.trace"
@@ -84,7 +83,6 @@ def run_benchmarks(
                 self.replay_tests_dir = Path(
                     tempfile.mkdtemp(prefix="codeflash_replay_tests_", dir=self.args.benchmarks_root)
                 )
-                logger.info(f"Tracing benchmarks to {trace_file} in {self.replay_tests_dir}")
                 trace_benchmarks_pytest(
                     self.args.benchmarks_root, self.args.tests_root, self.args.project_root, trace_file
                 )  # Run all tests that use pytest-benchmark