format and lint all

Author: misrasaurabh1

codeflash/api/aiservice.py

@@ -129,7 +129,8 @@ def optimize_code(  # noqa: D417
         experiment_metadata: ExperimentMetadata | None = None,
         *,
         language: str = "python",
-        language_version: str | None = None, # TODO:{claude} add language version to the language support and it should be cached
+        language_version: str
+        | None = None,  # TODO:{claude} add language version to the language support and it should be cached
         module_system: str | None = None,
         is_async: bool = False,
         n_candidates: int = 5,
@@ -238,6 +239,7 @@ def optimize_python_code(
             is_async=is_async,
             n_candidates=n_candidates,
         )
+
     def get_jit_rewritten_code(  # noqa: D417
         self, source_code: str, trace_id: str
     ) -> list[OptimizedCandidate]:
@@ -410,6 +412,7 @@ def optimize_code_refinement(self, request: list[AIServiceRefinerRequest]) -> li
 
         Returns:
             List of refined optimization candidates
+
         """
         payload = []
         for opt in request:
@@ -727,7 +730,7 @@ def generate_regression_tests(  # noqa: D417
         language: str = "python",
         language_version: str | None = None,
         module_system: str | None = None,
-        is_numerical_code: bool | None = None,  # noqa: FBT001
+        is_numerical_code: bool | None = None,
     ) -> tuple[str, str, str] | None:
         """Generate regression tests for the given function by making a request to the Django endpoint.
 

codeflash/api/schemas.py

@@ -14,7 +14,6 @@
 
 from dataclasses import dataclass, field
 from enum import Enum
-from pathlib import Path
 from typing import Any
 
 
@@ -192,10 +191,7 @@ def to_payload(self) -> dict[str, Any]:
         """Convert to API payload dict, maintaining backward compatibility."""
         payload = {
             "source_code_being_tested": self.source_code,
-            "function_to_optimize": {
-                "function_name": self.function_name,
-                "is_async": self.is_async,
-            },
+            "function_to_optimize": {"function_name": self.function_name, "is_async": self.is_async},
             "helper_function_names": self.helper_function_names,
             "module_path": self.module_path,
             "test_module_path": self.test_module_path,
@@ -243,24 +239,16 @@ def python_language_info(version: str | None = None) -> LanguageInfo:
 
 
 def javascript_language_info(
-    module_system: ModuleSystem = ModuleSystem.COMMONJS,
-    version: str = "ES2022",
+    module_system: ModuleSystem = ModuleSystem.COMMONJS, version: str = "ES2022"
 ) -> LanguageInfo:
     """Create LanguageInfo for JavaScript."""
     ext = ".mjs" if module_system == ModuleSystem.ESM else ".js"
     return LanguageInfo(
-        name="javascript",
-        version=version,
-        module_system=module_system,
-        file_extension=ext,
-        has_type_annotations=False,
+        name="javascript", version=version, module_system=module_system, file_extension=ext, has_type_annotations=False
     )
 
 
-def typescript_language_info(
-    module_system: ModuleSystem = ModuleSystem.ESM,
-    version: str = "ES2022",
-) -> LanguageInfo:
+def typescript_language_info(module_system: ModuleSystem = ModuleSystem.ESM, version: str = "ES2022") -> LanguageInfo:
     """Create LanguageInfo for TypeScript."""
     return LanguageInfo(
         name="typescript",

codeflash/cli_cmds/cmd_init.py

@@ -1516,6 +1516,7 @@ def _customize_python_workflow_content(
         codeflash_cmd += " --benchmark"
     return optimize_yml_content.replace("{{ codeflash_command }}", codeflash_cmd)
 
+
 # TODO:{claude} Refactor and move to support for language specific
 def _customize_js_workflow_content(
     optimize_yml_content: str,

codeflash/cli_cmds/console.py

@@ -108,6 +108,7 @@ def code_print(
         function_name: Optional function name for LSP
         lsp_message_id: Optional LSP message ID
         language: Programming language for syntax highlighting ('python', 'javascript', 'typescript')
+
     """
     if is_LSP_enabled():
         lsp_log(
@@ -118,11 +119,7 @@ def code_print(
     from rich.syntax import Syntax
 
     # Map codeflash language names to rich/pygments lexer names
-    lexer_map = {
-        "python": "python",
-        "javascript": "javascript",
-        "typescript": "typescript",
-    }
+    lexer_map = {"python": "python", "javascript": "javascript", "typescript": "typescript"}
     lexer = lexer_map.get(language, "python")
 
     console.rule()

codeflash/cli_cmds/init_javascript.py

@@ -1,4 +1,5 @@
 """JavaScript/TypeScript project initialization for Codeflash."""
+
 # TODO:{claude} move to language support directory
 from __future__ import annotations
 
@@ -125,8 +126,7 @@ def init_js_project(language: ProjectLanguage) -> None:
 
     lang_panel = Panel(
         Text(
-            f"📦 Detected {lang_name} project!\n\n"
-            "I'll help you set up Codeflash for your project.",
+            f"📦 Detected {lang_name} project!\n\nI'll help you set up Codeflash for your project.",
             style="cyan",
             justify="center",
         ),
@@ -159,13 +159,13 @@ def init_js_project(language: ProjectLanguage) -> None:
     usage_table.add_column("Command", style="cyan")
     usage_table.add_column("Description", style="white")
 
-    usage_table.add_row(
-        "codeflash --file <path-to-file> --function <function-name>", "Optimize a specific function"
-    )
+    usage_table.add_row("codeflash --file <path-to-file> --function <function-name>", "Optimize a specific function")
     usage_table.add_row("codeflash --all", "Optimize all functions in all files")
     usage_table.add_row("codeflash --help", "See all available options")
 
-    completion_message = f"⚡️ Codeflash is now set up for your {lang_name} project!\n\nYou can now run any of these commands:"
+    completion_message = (
+        f"⚡️ Codeflash is now set up for your {lang_name} project!\n\nYou can now run any of these commands:"
+    )
 
     if did_add_new_key:
         completion_message += (
@@ -265,10 +265,7 @@ def collect_js_setup_info(language: ProjectLanguage) -> JSSetupInfo:
     detection_table.add_row("Formatter", formatter_display)
 
     detection_panel = Panel(
-        Group(
-            Text(f"Auto-detected settings for your {lang_name} project:\n", style="cyan"),
-            detection_table,
-        ),
+        Group(Text(f"Auto-detected settings for your {lang_name} project:\n", style="cyan"), detection_table),
         title="🔍 Auto-Detection Results",
         border_style="bright_blue",
     )
@@ -399,8 +396,7 @@ def _get_git_remote_for_setup() -> str:
 
         git_panel = Panel(
             Text(
-                "🔗 Configure Git Remote for Pull Requests.\n\n"
-                "Codeflash will use this remote to create pull requests.",
+                "🔗 Configure Git Remote for Pull Requests.\n\nCodeflash will use this remote to create pull requests.",
                 style="blue",
             ),
             title="🔗 Git Remote Setup",

codeflash/code_utils/config_js.py

@@ -206,6 +206,7 @@ def find_package_json(config_file: Path | None = None) -> Path | None:
 
     return None
 
+
 def parse_package_json_config(package_json_path: Path) -> tuple[dict[str, Any], Path] | None:
     """Parse codeflash config from package.json with auto-detection.
 

codeflash/code_utils/deduplicate_code.py

@@ -8,14 +8,10 @@
 
 import hashlib
 import re
-from typing import TYPE_CHECKING
 
 from codeflash.code_utils.normalizers import get_normalizer
 from codeflash.languages import current_language, is_python
 
-if TYPE_CHECKING:
-    pass
-
 
 def normalize_code(
     code: str,
@@ -35,6 +31,7 @@ def normalize_code(
 
     Returns:
         Normalized code as string
+
     """
     if language is None:
         language = current_language().value
@@ -89,6 +86,7 @@ def get_code_fingerprint(code: str, language: str | None = None) -> str:
 
     Returns:
         SHA-256 hash of normalized code
+
     """
     if language is None:
         language = current_language().value
@@ -112,6 +110,7 @@ def are_codes_duplicate(code1: str, code2: str, language: str | None = None) ->
 
     Returns:
         True if codes are structurally identical (ignoring local variable names)
+
     """
     if language is None:
         language = current_language().value
@@ -127,8 +126,4 @@ def are_codes_duplicate(code1: str, code2: str, language: str | None = None) ->
 
 
 # Re-export for backward compatibility
-__all__ = [
-    "normalize_code",
-    "get_code_fingerprint",
-    "are_codes_duplicate",
-]
+__all__ = ["are_codes_duplicate", "get_code_fingerprint", "normalize_code"]

codeflash/code_utils/normalizers/__init__.py

@@ -22,13 +22,10 @@
 from codeflash.code_utils.normalizers.javascript import JavaScriptNormalizer, TypeScriptNormalizer
 from codeflash.code_utils.normalizers.python import PythonNormalizer
 
-if TYPE_CHECKING:
-    pass
-
 __all__ = [
     "CodeNormalizer",
-    "PythonNormalizer",
     "JavaScriptNormalizer",
+    "PythonNormalizer",
     "TypeScriptNormalizer",
     "get_normalizer",
     "get_normalizer_for_extension",
@@ -56,6 +53,7 @@ def get_normalizer(language: str) -> CodeNormalizer:
 
     Raises:
         ValueError: If no normalizer exists for the language
+
     """
     language = language.lower()
 
@@ -83,6 +81,7 @@ def get_normalizer_for_extension(extension: str) -> CodeNormalizer | None:
 
     Returns:
         CodeNormalizer instance if found, None otherwise
+
     """
     extension = extension.lower()
     if not extension.startswith("."):
@@ -102,6 +101,7 @@ def register_normalizer(language: str, normalizer_class: type[CodeNormalizer]) -
     Args:
         language: Language name
         normalizer_class: CodeNormalizer subclass
+
     """
     _NORMALIZERS[language.lower()] = normalizer_class
     # Clear cached instance if it exists

codeflash/code_utils/normalizers/base.py

@@ -4,14 +4,11 @@
 They normalize variable names, remove comments/docstrings, and produce consistent output
 that can be compared across different implementations of the same algorithm.
 """
+
 # TODO:{claude} move to base.py in language folder
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    pass
 
 
 class CodeNormalizer(ABC):
@@ -30,6 +27,7 @@ class CodeNormalizer(ABC):
         >>> code2 = "def foo(x): z = x + 1; return z"
         >>> normalizer.normalize(code1) == normalizer.normalize(code2)
         True
+
     """
 
     @property
@@ -52,6 +50,7 @@ def normalize(self, code: str) -> str:
 
         Returns:
             Normalized representation of the code
+
         """
         ...
 
@@ -66,6 +65,7 @@ def normalize_for_hash(self, code: str) -> str:
 
         Returns:
             Normalized representation suitable for hashing
+
         """
         ...
 
@@ -78,6 +78,7 @@ def are_duplicates(self, code1: str, code2: str) -> bool:
 
         Returns:
             True if codes are structurally identical
+
         """
         try:
             normalized1 = self.normalize_for_hash(code1)
@@ -94,6 +95,7 @@ def get_fingerprint(self, code: str) -> str:
 
         Returns:
             SHA-256 hash of normalized code
+
         """
         import hashlib