Merge pull request #1590 from codeflash-ai/codeflash/optimize-pr1199-2026-02-20T07.48.46

⚡️ Speed up function `_infer_array_cast_type` by 81% in PR #1199 (`omni-java`)

Author: claude[bot]

codeflash/languages/java/instrumentation.py

@@ -259,11 +259,11 @@ def _infer_array_cast_type(line: str) -> str | None:
 
     # Look for primitive array type in the line (usually the first/expected argument)
     match = _PRIMITIVE_ARRAY_PATTERN.search(line)
-    if match:
-        primitive_type = match.group(1)
-        return f"{primitive_type}[]"
+    if not match:
+        return None
 
-    return None
+    primitive_type = match.group(1)
+    return f"{primitive_type}[]"
 
 
 def _get_qualified_name(func: Any) -> str: