⚡️ Speed up method BenchmarkFunctionRemover._is_benchmark_call by 16% in PR #313 (skip-benchmark-instrumentation)

Here’s an optimized version of your program, designed to improve runtime **without** altering the function signatures or output values. The function is quite simple, but we streamline the checks for speed and minimize unnecessary computation by.

- Saving attribute lookups to variables to avoid recalculating them multiple times.
- Reordering `if` checks so that the least expensive checks are performed first.
- Returning early to avoid the cost of forming an unnecessary `bool()` call or evaluating unnecessary conditions.



**Preserved all comments.**  
This removes a function call to `bool`, skips unnecessary conditions, and avoids both repeated lookups and unnecessary object creation.  
The logic is unchanged; this is strictly faster.

Author: codeflash-ai[bot]

codeflash/code_utils/code_replacer.py

@@ -76,14 +76,15 @@ def _is_benchmark_marker(self, decorator: ast.expr) -> bool:
     @staticmethod
     def _is_benchmark_call(call: ast.Call) -> bool:
         """Check if a call is using the benchmark fixture."""
-        if isinstance(call.func, ast.Name) and call.func.id == "benchmark":
-            return True
-        return bool(
-            isinstance(call.func, ast.Attribute)
-            and call.func.attr in ["benchmark", "__call__"]
-            and isinstance(call.func.value, ast.Name)
-            and call.func.value.id == "benchmark"
-        )
+        func = call.func
+        if isinstance(func, ast.Name):
+            return func.id == "benchmark"
+        if isinstance(func, ast.Attribute):
+            if func.attr not in {"benchmark", "__call__"}:
+                return False
+            value = func.value
+            return isinstance(value, ast.Name) and value.id == "benchmark"
+        return False
 
     def visit_FunctionDef(self, node: ast.FunctionDef) -> Optional[AST]:
         """Visit function definitions and remove if they use benchmark fixture."""