formatting

Author: KRRT7

codeflash/cli_cmds/cli.py

@@ -352,7 +352,7 @@ def _handle_show_config() -> None:
     from codeflash.setup.detector import detect_project, has_existing_config
 
     project_root = Path.cwd()
-    config_exists, config_file = has_existing_config(project_root)
+    config_exists, _ = has_existing_config(project_root)
 
     if config_exists:
         from codeflash.code_utils.config_parser import parse_config_file
@@ -373,10 +373,7 @@ def _handle_show_config() -> None:
         table.add_row("Module root", config.get("module_root", "(not set)"))
         table.add_row("Tests root", config.get("tests_root", "(not set)"))
         table.add_row("Test runner", config.get("test_framework", config.get("pytest_cmd", "(not set)")))
-        table.add_row(
-            "Formatter",
-            ", ".join(config["formatter_cmds"]) if config.get("formatter_cmds") else "(not set)",
-        )
+        table.add_row("Formatter", ", ".join(config["formatter_cmds"]) if config.get("formatter_cmds") else "(not set)")
         ignore_paths = config.get("ignore_paths", [])
         table.add_row("Ignore paths", ", ".join(str(p) for p in ignore_paths) if ignore_paths else "(none)")
     else:
@@ -396,12 +393,9 @@ def _handle_show_config() -> None:
         table.add_row("Module root", str(detected.module_root))
         table.add_row("Tests root", str(detected.tests_root) if detected.tests_root else "(not detected)")
         table.add_row("Test runner", detected.test_runner or "(not detected)")
+        table.add_row("Formatter", ", ".join(detected.formatter_cmds) if detected.formatter_cmds else "(not detected)")
         table.add_row(
-            "Formatter", ", ".join(detected.formatter_cmds) if detected.formatter_cmds else "(not detected)"
-        )
-        table.add_row(
-            "Ignore paths",
-            ", ".join(str(p) for p in detected.ignore_paths) if detected.ignore_paths else "(none)",
+            "Ignore paths", ", ".join(str(p) for p in detected.ignore_paths) if detected.ignore_paths else "(none)"
         )
         table.add_row("Confidence", f"{detected.confidence:.0%}")
 

codeflash/code_utils/time_utils.py

@@ -98,4 +98,6 @@ def format_runtime_comment(original_time_ns: int, optimized_time_ns: int, commen
         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})"
+    return (
+        f"{comment_prefix} {format_time(original_time_ns)} -> {format_time(optimized_time_ns)} ({perf_gain}% {status})"
+    )

codeflash/languages/python/context/code_context_extractor.py

@@ -23,7 +23,10 @@
     recurse_sections,
     remove_unused_definitions_by_function_names,
 )
-from codeflash.languages.python.static_analysis.code_extractor import add_needed_imports_from_module, find_preexisting_objects
+from codeflash.languages.python.static_analysis.code_extractor import (
+    add_needed_imports_from_module,
+    find_preexisting_objects,
+)
 from codeflash.models.models import (
     CodeContextType,
     CodeOptimizationContext,

codeflash/languages/python/static_analysis/code_extractor.py

@@ -1113,9 +1113,7 @@ def find_target(node_list: list[ast.stmt], name_parts: tuple[str, str] | tuple[s
         target: ast.FunctionDef | ast.AsyncFunctionDef | ast.ClassDef | ast.Assign | ast.AnnAssign | None = None
         node: ast.stmt
         for node in node_list:
-            if (
-                isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef)) and node.name == name_parts[0]
-            ):
+            if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef)) and node.name == name_parts[0]:
                 target = node
                 break
             if (

codeflash/optimization/function_optimizer.py

@@ -71,15 +71,13 @@
     revert_unused_helper_functions,
 )
 from codeflash.languages.python.static_analysis._ast import get_first_top_level_function_or_method_ast
-from codeflash.languages.python.static_analysis.code_extractor import is_numerical_code
-from codeflash.languages.python.static_analysis.line_profile_utils import contains_jit_decorator
-from codeflash.languages.python.static_analysis.code_extractor import get_opt_review_metrics
+from codeflash.languages.python.static_analysis.code_extractor import get_opt_review_metrics, is_numerical_code
 from codeflash.languages.python.static_analysis.code_replacer import (
     add_custom_marker_to_all_tests,
     modify_autouse_fixture,
     replace_function_definitions_in_module,
 )
-from codeflash.languages.python.static_analysis.line_profile_utils import add_decorator_imports
+from codeflash.languages.python.static_analysis.line_profile_utils import add_decorator_imports, contains_jit_decorator
 from codeflash.lsp.helpers import is_LSP_enabled, report_to_markdown_table, tree_to_markdown
 from codeflash.lsp.lsp_message import LspCodeMessage, LspMarkdownMessage, LSPMessageId
 from codeflash.models.ExperimentMetadata import ExperimentMetadata

codeflash/verification/coverage_utils.py

@@ -7,8 +7,11 @@
 from coverage.exceptions import NoDataError
 
 from codeflash.cli_cmds.console import logger
-from codeflash.languages.python.static_analysis.coverage_utils import extract_dependent_function
-from codeflash.languages.python.static_analysis.coverage_utils import build_fully_qualified_name, generate_candidates
+from codeflash.languages.python.static_analysis.coverage_utils import (
+    build_fully_qualified_name,
+    extract_dependent_function,
+    generate_candidates,
+)
 from codeflash.models.models import CoverageData, CoverageStatus, FunctionCoverage
 
 if TYPE_CHECKING: