restore plugin

Author: KRRT7

codeflash/benchmarking/pytest_new_process_trace_benchmarks.py

@@ -1,8 +1,8 @@
 import sys
 from pathlib import Path
 
-# from codeflash.benchmarking.codeflash_trace import codeflash_trace
-# from codeflash.benchmarking.plugin.plugin import codeflash_benchmark_plugin
+from codeflash.benchmarking.codeflash_trace import codeflash_trace
+from codeflash.benchmarking.plugin.plugin import codeflash_benchmark_plugin
 
 benchmarks_root = sys.argv[1]
 tests_root = sys.argv[2]
@@ -17,8 +17,8 @@
         original_limit = sys.getrecursionlimit()
         sys.setrecursionlimit(max(2000, original_limit * 2))
 
-        # codeflash_benchmark_plugin.setup(trace_file, project_root)
-        # codeflash_trace.setup(trace_file)
+        codeflash_benchmark_plugin.setup(trace_file, project_root)
+        codeflash_trace.setup(trace_file)
         exitcode = pytest.main(
             [
                 benchmarks_root,
@@ -35,7 +35,7 @@
                 "-o",
                 "addopts=",
             ],
-            # plugins=[codeflash_benchmark_plugin],
+            plugins=[codeflash_benchmark_plugin],
         )  # Errors will be printed to stdout, not stderr
 
         sys.setrecursionlimit(original_limit)