Update plugin.py

Author: KRRT7

codeflash/benchmarking/plugin/plugin.py

@@ -52,7 +52,7 @@ def write_benchmark_timings(self) -> None:
         try:
             cur = self._connection.cursor()
             cur.executemany(
-                "INSERT INTO benchmark_timings (benchmark_module_path, benchmark_function_name, benchmark_line_number, benchmark_time_ns) VALUES (?, ?, ?, ?)",
+                "INSERT INTO benchmark_timings (benchmark_module_path, benchmark_function_name, benchmark_line_number, benchmark_time_ns) VALUES (?, ?, ?, ?)",  # noqa: E501
                 self.benchmark_timings,
             )
             self._connection.commit()
@@ -238,7 +238,7 @@ def _run_benchmark(
                 Path(str(self.request.node.fspath)), Path(codeflash_benchmark_plugin.project_root)
             )
             benchmark_function_name = self.request.node.name
-            line_number = int(str(sys._getframe(2).f_lineno))  # 2 frames up in the call stack
+            line_number = int(str(sys._getframe(2).f_lineno))  # 2 frames up in the call stack  # noqa: SLF001
             self._call_count += 1
             if benchmark_name_suffix:
                 call_identifier = f"{benchmark_function_name}::{benchmark_name_suffix}"
@@ -260,7 +260,9 @@ def _run_benchmark(
             codeflash_benchmark_plugin.benchmark_timings.append(
                 (benchmark_module_path, call_identifier, line_number, end - start)
             )
-
+            logger.info(
+                f"Benchmarking {benchmark_module_path}::{call_identifier} took {(end - start) / 1_000_000:.2f} ms"
+            )
             return result
 
     @staticmethod