style: auto-fix linting and formatting issues

Author: github-actions[bot]

codeflash/cli_cmds/console.py

@@ -40,7 +40,16 @@
 
 logging.basicConfig(
     level=logging.INFO,
-    handlers=[RichHandler(rich_tracebacks=True, markup=False, highlighter=NullHighlighter(), console=console, show_path=False, show_time=False)],
+    handlers=[
+        RichHandler(
+            rich_tracebacks=True,
+            markup=False,
+            highlighter=NullHighlighter(),
+            console=console,
+            show_path=False,
+            show_time=False,
+        )
+    ],
     format=BARE_LOGGING_FORMAT,
 )
 

codeflash/cli_cmds/logging_config.py

@@ -14,7 +14,16 @@ def set_level(level: int, *, echo_setting: bool = True) -> None:
 
     logging.basicConfig(
         level=level,
-        handlers=[RichHandler(rich_tracebacks=True, markup=False, highlighter=NullHighlighter(), console=console, show_path=False, show_time=False)],
+        handlers=[
+            RichHandler(
+                rich_tracebacks=True,
+                markup=False,
+                highlighter=NullHighlighter(),
+                console=console,
+                show_path=False,
+                show_time=False,
+            )
+        ],
         format=BARE_LOGGING_FORMAT,
     )
     logging.getLogger().setLevel(level)
@@ -23,7 +32,14 @@ def set_level(level: int, *, echo_setting: bool = True) -> None:
         logging.basicConfig(
             format=VERBOSE_LOGGING_FORMAT,
             handlers=[
-                RichHandler(rich_tracebacks=True, markup=False, highlighter=NullHighlighter(), console=console, show_path=False, show_time=False)
+                RichHandler(
+                    rich_tracebacks=True,
+                    markup=False,
+                    highlighter=NullHighlighter(),
+                    console=console,
+                    show_path=False,
+                    show_time=False,
+                )
             ],
             force=True,
         )

codeflash/languages/java/context.py

@@ -887,11 +887,7 @@ def collect_type_identifiers(node: Node) -> None:
 
 
 def get_java_imported_type_skeletons(
-    imports: list,
-    project_root: Path,
-    module_root: Path | None,
-    analyzer: JavaAnalyzer,
-    target_code: str = "",
+    imports: list, project_root: Path, module_root: Path | None, analyzer: JavaAnalyzer, target_code: str = ""
 ) -> str:
     """Extract type skeletons for project-internal imported types.
 
@@ -1011,9 +1007,7 @@ def _extract_constructor_summaries(skeleton: TypeSkeleton) -> list[str]:
     return summaries
 
 
-def _format_skeleton_for_context(
-    skeleton: TypeSkeleton, source: str, class_name: str, analyzer: JavaAnalyzer
-) -> str:
+def _format_skeleton_for_context(skeleton: TypeSkeleton, source: str, class_name: str, analyzer: JavaAnalyzer) -> str:
     """Format a TypeSkeleton into a context string with method signatures.
 
     Includes: type declaration, fields, constructors, and public method signatures
@@ -1094,7 +1088,7 @@ def _extract_public_method_signatures(source: str, class_name: str, analyzer: Ja
                 sig_parts_bytes.append(mod_slice)
                 continue
 
-            if ctype == "block" or ctype == "constructor_body":
+            if ctype in {"block", "constructor_body"}:
                 break
 
             sig_parts_bytes.append(source_bytes[child.start_byte : child.end_byte])

codeflash/languages/java/instrumentation.py

@@ -730,11 +730,17 @@ def split_var_declaration(stmt_node, source_bytes_ref: bytes) -> tuple[str, str]
         # The variable is assigned inside a for/try block which Java considers
         # conditionally executed, so an uninitialized declaration would cause
         # "variable might not have been initialized" errors.
-        _PRIMITIVE_DEFAULTS = {
-            "byte": "0", "short": "0", "int": "0", "long": "0L",
-            "float": "0.0f", "double": "0.0", "char": "'\\0'", "boolean": "false",
+        primitive_defaults = {
+            "byte": "0",
+            "short": "0",
+            "int": "0",
+            "long": "0L",
+            "float": "0.0f",
+            "double": "0.0",
+            "char": "'\\0'",
+            "boolean": "false",
         }
-        default_val = _PRIMITIVE_DEFAULTS.get(type_text, "null")
+        default_val = primitive_defaults.get(type_text, "null")
         hoisted = f"{type_text} {name_text} = {default_val};"
         assignment = f"{name_text} = {value_text};"
         return hoisted, assignment
@@ -918,9 +924,7 @@ def build_instrumented_body(body_text: str, next_wrapper_id: int, base_indent: s
 
     replacements: list[tuple[int, int, bytes]] = []
     wrapper_id = 0
-    method_ordinal = 0
-    for method_node, body_node in test_methods:
-        method_ordinal += 1
+    for method_ordinal, (method_node, body_node) in enumerate(test_methods, start=1):
         body_start = body_node.start_byte + 1  # skip '{'
         body_end = body_node.end_byte - 1  # skip '}'
         body_text = source_bytes[body_start:body_end].decode("utf8")

codeflash/languages/java/replacement.py

@@ -374,13 +374,7 @@ def replace_function(
                 class_name,
             )
             source = _insert_class_members(
-                source,
-                class_name,
-                new_fields_to_add,
-                new_helpers_before,
-                new_helpers_after,
-                func_name,
-                analyzer,
+                source, class_name, new_fields_to_add, new_helpers_before, new_helpers_after, func_name, analyzer
             )
 
             # Re-find the target method after modifications

codeflash/verification/parse_line_profile_test_output.py

@@ -6,16 +6,14 @@
 import json
 import linecache
 import os
-from typing import TYPE_CHECKING, Optional
+from pathlib import Path
+from typing import Optional
 
 import dill as pickle
 
 from codeflash.code_utils.tabulate import tabulate
 from codeflash.languages import is_python
 
-if TYPE_CHECKING:
-    from pathlib import Path
-
 
 def show_func(
     filename: str, start_lineno: int, func_name: str, timings: list[tuple[int, int, float]], unit: float
@@ -80,9 +78,7 @@ def show_text(stats: dict) -> str:
     return out_table
 
 
-def show_text_non_python(
-    stats: dict, line_contents: dict[tuple[str, int], str]
-) -> str:
+def show_text_non_python(stats: dict, line_contents: dict[tuple[str, int], str]) -> str:
     """Show text for non-Python timings using profiler-provided line contents."""
     out_table = ""
     out_table += "# Timer unit: {:g} s\n".format(stats["unit"])
@@ -100,13 +96,13 @@ def show_text_non_python(
         table_rows = []
         for lineno, nhits, time in timings:
             percent = "" if total_time == 0 else "%5.1f" % (100 * time / total_time)
-            time_disp = "%5.1f" % time
+            time_disp = f"{time:5.1f}"
             if len(time_disp) > default_column_sizes["time"]:
-                time_disp = "%5.1g" % time
+                time_disp = f"{time:5.1g}"
             perhit = (float(time) / nhits) if nhits > 0 else 0.0
-            perhit_disp = "%5.1f" % perhit
+            perhit_disp = f"{perhit:5.1f}"
             if len(perhit_disp) > default_column_sizes["perhit"]:
-                perhit_disp = "%5.1g" % perhit
+                perhit_disp = f"{perhit:5.1g}"
             nhits_disp = "%d" % nhits  # noqa: UP031
             if len(nhits_disp) > default_column_sizes["hits"]:
                 nhits_disp = f"{nhits:g}"
@@ -115,11 +111,7 @@ def show_text_non_python(
 
         table_cols = ("Hits", "Time", "Per Hit", "% Time", "Line Contents")
         out_table += tabulate(
-            headers=table_cols,
-            tabular_data=table_rows,
-            tablefmt="pipe",
-            colglobalalign=None,
-            preserve_whitespace=True,
+            headers=table_cols, tabular_data=table_rows, tablefmt="pipe", colglobalalign=None, preserve_whitespace=True
         )
         out_table += "\n"
     return out_table
@@ -159,17 +151,15 @@ def parse_line_profile_results(line_profiler_output_file: Optional[Path]) -> dic
             line_num = int(line_str)
         line_num = int(line_num)
 
-        lines_by_file.setdefault(file_path, []).append(
-            (line_num, int(stats.get("hits", 0)), int(stats.get("time", 0)))
-        )
+        lines_by_file.setdefault(file_path, []).append((line_num, int(stats.get("hits", 0)), int(stats.get("time", 0))))
         line_contents[(file_path, line_num)] = stats.get("content", "")
 
     for file_path, line_stats in lines_by_file.items():
         sorted_line_stats = sorted(line_stats, key=lambda t: t[0])
         if not sorted_line_stats:
             continue
         start_lineno = sorted_line_stats[0][0]
-        grouped_timings[(file_path, start_lineno, os.path.basename(file_path))] = sorted_line_stats
+        grouped_timings[(file_path, start_lineno, Path(file_path).name)] = sorted_line_stats
 
     stats_dict["timings"] = grouped_timings
     stats_dict["unit"] = 1e-9