Update pytest_new_process_trace_benchmarks.py

Author: KRRT7

codeflash/benchmarking/pytest_new_process_trace_benchmarks.py

@@ -2,7 +2,6 @@
 from pathlib import Path
 
 import pytest
-pytest.register_assert_rewrite("codeflash")
 
 benchmarks_root = sys.argv[1]
 tests_root = sys.argv[2]
@@ -11,8 +10,8 @@
 project_root = Path.cwd()
 
 if __name__ == "__main__":
-    from codeflash.benchmarking.codeflash_trace import codeflash_trace
-    from codeflash.benchmarking.plugin.plugin import codeflash_benchmark_plugin
+    from benchmarking.codeflash_trace import codeflash_trace
+    from benchmarking.plugin.plugin import codeflash_benchmark_plugin
 
     try:
         # Increase recursion limit to handle recursive benchmarks with tracing overhead