Merge pull request #1145 from codeflash-ai/jit-docs

Documentation for JIT support

Author: aseembits93

codeflash/benchmarking/plugin/plugin.py

@@ -200,7 +200,7 @@ def get_benchmark_timings(trace_path: Path) -> dict[BenchmarkKey, int]:
 
     # Pytest hooks
     @pytest.hookimpl
-    def pytest_sessionfinish(self, session, exitstatus) -> None:  # noqa: ANN001
+    def pytest_sessionfinish(self, session, exitstatus) -> None:
         """Execute after whole test run is completed."""
         # Write any remaining benchmark timings to the database
         codeflash_trace.close()
@@ -236,20 +236,20 @@ class Benchmark:  # noqa: D106
         def __init__(self, request: pytest.FixtureRequest) -> None:
             self.request = request
 
-        def __call__(self, func, *args, **kwargs):  # noqa: ANN001, ANN002, ANN003, ANN204
+        def __call__(self, func, *args, **kwargs):  # noqa: ANN002, ANN003, ANN204
             """Handle both direct function calls and decorator usage."""
             if args or kwargs:
                 # Used as benchmark(func, *args, **kwargs)
                 return self._run_benchmark(func, *args, **kwargs)
 
             # Used as @benchmark decorator
-            def wrapped_func(*args, **kwargs):  # noqa: ANN002, ANN003, ANN202
+            def wrapped_func(*args, **kwargs):  # noqa: ANN002, ANN003
                 return func(*args, **kwargs)
 
             self._run_benchmark(func)
             return wrapped_func
 
-        def _run_benchmark(self, func, *args, **kwargs):  # noqa: ANN002, ANN003, ANN202
+        def _run_benchmark(self, func, *args, **kwargs):  # noqa: ANN002, ANN003
             """Actual benchmark implementation."""
             node_path = getattr(self.request.node, "path", None) or getattr(self.request.node, "fspath", None)
             if node_path is None:

codeflash/cli_cmds/cmd_init.py

@@ -1474,11 +1474,7 @@ def customize_codeflash_yaml_content(
     return _customize_python_workflow_content(optimize_yml_content, git_root, benchmark_mode)
 
 
-def _customize_python_workflow_content(
-    optimize_yml_content: str,
-    git_root: Path,
-    benchmark_mode: bool = False,  # noqa: FBT001, FBT002
-) -> str:
+def _customize_python_workflow_content(optimize_yml_content: str, git_root: Path, benchmark_mode: bool = False) -> str:
     """Customize workflow content for Python projects."""
     # Get dependency installation commands
     toml_path = Path.cwd() / "pyproject.toml"
@@ -1513,11 +1509,7 @@ def _customize_python_workflow_content(
 
 
 # TODO:{claude} Refactor and move to support for language specific
-def _customize_js_workflow_content(
-    optimize_yml_content: str,
-    git_root: Path,
-    benchmark_mode: bool = False,  # noqa: FBT001, FBT002
-) -> str:
+def _customize_js_workflow_content(optimize_yml_content: str, git_root: Path, benchmark_mode: bool = False) -> str:
     """Customize workflow content for JavaScript/TypeScript projects."""
     from codeflash.cli_cmds.init_javascript import (
         get_js_codeflash_install_step,

codeflash/cli_cmds/init_javascript.py

@@ -66,7 +66,7 @@ class JSSetupInfo:
 
 
 # Import theme from cmd_init to avoid duplication
-def _get_theme():  # noqa: ANN202
+def _get_theme():
     """Get the CodeflashTheme - imported lazily to avoid circular imports."""
     from codeflash.cli_cmds.cmd_init import CodeflashTheme
 

codeflash/code_utils/code_utils.py

@@ -436,7 +436,7 @@ def extract_unique_errors(pytest_output: str) -> set[str]:
     pattern = r"^E\s+(.*)$"
 
     for error_message in re.findall(pattern, pytest_output, re.MULTILINE):
-        error_message = error_message.strip()  # noqa: PLW2901
+        error_message = error_message.strip()
         if error_message:
             unique_errors.add(error_message)
 

codeflash/code_utils/config_js.py

@@ -105,7 +105,7 @@ def detect_module_root(project_root: Path, package_data: dict[str, Any]) -> str:
     return "."
 
 
-def detect_test_runner(project_root: Path, package_data: dict[str, Any]) -> str:  # noqa: ARG001
+def detect_test_runner(project_root: Path, package_data: dict[str, Any]) -> str:
     """Detect test runner from devDependencies or scripts.test.
 
     Detection order:
@@ -144,7 +144,7 @@ def detect_test_runner(project_root: Path, package_data: dict[str, Any]) -> str:
     return "jest"
 
 
-def detect_formatter(project_root: Path, package_data: dict[str, Any]) -> list[str] | None:  # noqa: ARG001
+def detect_formatter(project_root: Path, package_data: dict[str, Any]) -> list[str] | None:
     """Detect formatter from devDependencies.
 
     Detection order:

codeflash/code_utils/deduplicate_code.py

@@ -14,10 +14,7 @@
 
 
 def normalize_code(
-    code: str,
-    remove_docstrings: bool = True,
-    return_ast_dump: bool = False,
-    language: str | None = None,
+    code: str, remove_docstrings: bool = True, return_ast_dump: bool = False, language: str | None = None
 ) -> str:
     """Normalize code by parsing, cleaning, and normalizing variable names.
 

codeflash/code_utils/instrument_existing_tests.py

@@ -89,7 +89,7 @@ def find_and_update_line_node(
         # it's much more efficient to visit nodes manually. We'll only descend into expressions/statements.
 
         # Helper for manual walk
-        def iter_ast_calls(node):  # noqa: ANN202
+        def iter_ast_calls(node):
             # Generator to yield each ast.Call in test_node, preserves node identity
             stack = [node]
             while stack:
@@ -102,7 +102,7 @@ def iter_ast_calls(node):  # noqa: ANN202
                     if isinstance(value, list):
                         for item in reversed(value):
                             if isinstance(item, ast.AST):
-                                stack.append(item)  # noqa: PERF401
+                                stack.append(item)
                     elif isinstance(value, ast.AST):
                         stack.append(value)
 

codeflash/context/code_context_extractor.py

@@ -46,8 +46,8 @@ def build_testgen_context(
     helpers_of_fto_dict: dict[Path, set[FunctionSource]],
     helpers_of_helpers_dict: dict[Path, set[FunctionSource]],
     project_root_path: Path,
-    remove_docstrings: bool,  # noqa: FBT001
-    include_imported_classes: bool,  # noqa: FBT001
+    remove_docstrings: bool,
+    include_imported_classes: bool,
 ) -> CodeStringsMarkdown:
     """Build testgen context with optional imported class definitions and external base inits."""
     testgen_context = extract_code_markdown_context_from_files(

codeflash/languages/javascript/module_system.py

@@ -185,7 +185,7 @@ def _get_relative_import_path(target_path: Path, source_path: Path) -> str:
 
 def add_js_extension(module_path: str) -> str:
     """Add .js extension to relative module paths for ESM compatibility."""
-    if module_path.startswith(("./", "../")):  # noqa: SIM102
+    if module_path.startswith(("./", "../")):
         if not module_path.endswith(".js") and not module_path.endswith(".mjs"):
             return module_path + ".js"
     return module_path

codeflash/languages/javascript/support.py

@@ -1872,7 +1872,7 @@ def instrument_source_for_line_profiler(
             # Write instrumented code to source file
             source_file_path.write_text(instrumented_source, encoding="utf-8")
             logger.debug("Wrote instrumented source to %s", source_file_path)
-            return True  # noqa: TRY300
+            return True
         except Exception as e:
             logger.warning("Failed to instrument source for line profiling: %s", e)
             return False