⚡️ Speed up method PrComment.to_json by 28% in PR #1199 (omni-java)

codeflash/code_utils/time_utils.py

@@ -1,6 +1,11 @@
 from __future__ import annotations
 
+from functools import lru_cache
 
+from codeflash.result.critic import performance_gain
+
+
+@lru_cache(maxsize=1024)
 def humanize_runtime(time_in_ns: int) -> str:
     runtime_human: str = str(time_in_ns)
     units = "nanoseconds"
@@ -89,3 +94,13 @@ def format_perf(percentage: float) -> str:
     if abs_perc >= 1:
         return f"{percentage:.2f}"
     return f"{percentage:.3f}"
+
+
+def format_runtime_comment(original_time_ns: int, optimized_time_ns: int, comment_prefix: str = "#") -> str:
+    perf_gain = format_perf(
+        abs(performance_gain(original_runtime_ns=original_time_ns, optimized_runtime_ns=optimized_time_ns) * 100)
+    )
+    status = "slower" if optimized_time_ns > original_time_ns else "faster"
+    return (
+        f"{comment_prefix} {format_time(original_time_ns)} -> {format_time(optimized_time_ns)} ({perf_gain}% {status})"
+    )

codeflash/languages/__init__.py

@@ -38,7 +38,6 @@
     reset_current_language,
     set_current_language,
 )
-
 from codeflash.languages.registry import (
     detect_project_language,
     get_language_support,

codeflash/languages/registry.py

@@ -53,7 +53,7 @@ def _ensure_languages_registered() -> None:
         from codeflash.languages.python import support as _
 
     with contextlib.suppress(ImportError):
-        from codeflash.languages.javascript import support as _  # noqa: F401
+        from codeflash.languages.javascript import support as _
 
     with contextlib.suppress(ImportError):
         from codeflash.languages.java import support as _  # noqa: F401