ruff check .

codeflash/cli_cmds/cmd_init.py

@@ -597,7 +597,7 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
     if formatter in ["black", "ruff"]:
         try:
             result = subprocess.run([formatter], capture_output=True, check=False)
-        except FileNotFoundError as e:
+        except FileNotFoundError:
             click.echo(f"⚠️ Formatter not found: {formatter}, please ensure it is installed")
     codeflash_section["formatter-cmds"] = formatter_cmds
     # Add the 'codeflash' section, ensuring 'tool' section exists

codeflash/cli_cmds/console.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 import logging
+from collections.abc import Generator
 from contextlib import contextmanager
 from itertools import cycle
-from typing import TYPE_CHECKING, Generator
+from typing import TYPE_CHECKING
 
 from rich.console import Console
 from rich.logging import RichHandler

codeflash/code_utils/code_extractor.py

@@ -115,8 +115,8 @@ def get_code(functions_to_optimize: list[FunctionToOptimize]) -> tuple[str | Non
         or (functions_to_optimize[0].parents and functions_to_optimize[0].parents[0].type != "ClassDef")
         or (
             len(functions_to_optimize[0].parents) > 1
-            or (len(functions_to_optimize) > 1)
-            and len({fn.parents[0] for fn in functions_to_optimize}) != 1
+            or ((len(functions_to_optimize) > 1)
+            and len({fn.parents[0] for fn in functions_to_optimize}) != 1)
         )
     ):
         return None, set()

codeflash/code_utils/code_replacer.py

@@ -1,10 +1,9 @@
 from __future__ import annotations
 
 import ast
-import re
 from collections import defaultdict
 from functools import lru_cache
-from typing import TYPE_CHECKING, Optional, TypeVar
+from typing import TYPE_CHECKING, TypeVar
 
 import libcst as cst
 

codeflash/code_utils/compat.py

@@ -11,4 +11,4 @@ LF: str = os.linesep
 
 SAFE_SYS_EXECUTABLE: str = Path(sys.executable).as_posix()
 
-IS_POSIX = os.name != "nt"
+IS_POSIX = os.name != "nt"

codeflash/code_utils/formatter.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 import os
 import shlex
 import subprocess
-import sys
 from typing import TYPE_CHECKING
 
 import isort

codeflash/discovery/functions_to_optimize.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import ast
-import json
 import os
 import random
 import warnings

codeflash/optimization/function_optimizer.py

@@ -75,9 +75,6 @@ from codeflash.verification.verifier import generate_tests
 if TYPE_CHECKING:
     from argparse import Namespace
 
-    import numpy as np
-    import numpy.typing as npt
-
     from codeflash.either import Result
     from codeflash.models.models import CoverageData, FunctionSource, OptimizedCandidate
     from codeflash.verification.verification_utils import TestConfig

codeflash/optimization/optimizer.py

@@ -16,7 +16,7 @@ from codeflash.code_utils.static_analysis import analyze_imported_modules, get_f
 from codeflash.discovery.discover_unit_tests import discover_unit_tests
 from codeflash.discovery.functions_to_optimize import get_functions_to_optimize
 from codeflash.either import is_successful
-from codeflash.models.models import TestFiles, ValidCode
+from codeflash.models.models import ValidCode
 from codeflash.optimization.function_optimizer import FunctionOptimizer
 from codeflash.telemetry.posthog_cf import ph
 from codeflash.verification.test_results import TestType

codeflash/tracing/replay_test.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 import sqlite3
 import textwrap
-from typing import Any, Generator, List, Optional
+from collections.abc import Generator
+from typing import Any, List, Optional
 
 from codeflash.discovery.functions_to_optimize import FunctionProperties, inspect_top_level_functions_or_methods
 from codeflash.tracing.tracing_utils import FunctionModules

codeflash/update_license_version.py

@@ -1,4 +1,3 @@
-import os
 import re
 from datetime import datetime
 from pathlib import Path

codeflash/verification/concolic_testing.py

@@ -7,8 +7,8 @@ from argparse import Namespace
 from pathlib import Path
 
 from codeflash.cli_cmds.console import console, logger
-from codeflash.code_utils.concolic_utils import clean_concolic_tests
 from codeflash.code_utils.compat import SAFE_SYS_EXECUTABLE
+from codeflash.code_utils.concolic_utils import clean_concolic_tests
 from codeflash.code_utils.static_analysis import has_typed_parameters
 from codeflash.discovery.discover_unit_tests import discover_unit_tests
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize

tests/test_code_context_extractor.py

@@ -6,6 +6,7 @@ from collections import defaultdict
 from pathlib import Path
 
 import pytest
+
 from codeflash.context.code_context_extractor import get_code_optimization_context
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.models.models import FunctionParent

tests/test_comparator.py

@@ -10,7 +10,6 @@ from pathlib import Path
 
 import pydantic
 import pytest
-from pathlib import Path
 
 from codeflash.either import Failure, Success
 from codeflash.verification.comparator import comparator
@@ -302,7 +301,7 @@ def test_numpy():
 
 def test_scipy():
     try:
-        import scipy as sp # type: ignore
+        import scipy as sp  # type: ignore
     except ImportError:
         pytest.skip()
     a = sp.sparse.csr_matrix([[1, 0, 0], [0, 0, 3], [4, 0, 5]])
@@ -472,7 +471,7 @@ def test_pandas():
 
 def test_pyrsistent():
     try:
-        from pyrsistent import PBag, PClass, PRecord, field, pdeque, pmap, pset, pvector # type: ignore
+        from pyrsistent import PBag, PClass, PRecord, field, pdeque, pmap, pset, pvector  # type: ignore
     except ImportError:
         pytest.skip()
 
@@ -1039,7 +1038,7 @@ def test_exceptions_comparator():
 
     assert not comparator(..., None)
 
-    assert not comparator(Ellipsis, None)   
+    assert not comparator(Ellipsis, None)
 
     code7 = "a = 1 + 2"
     module7 = ast.parse(code7)
@@ -1053,4 +1052,4 @@ def test_exceptions_comparator():
 
     module2 = ast.parse(code2)
 
-    assert not comparator(module7, module2)
+    assert not comparator(module7, module2)

tests/test_formatter.py

@@ -3,6 +3,7 @@ import tempfile
 from pathlib import Path
 
 import pytest
+
 from codeflash.code_utils.config_parser import parse_config_file
 from codeflash.code_utils.formatter import format_code, sort_imports
 

tests/test_function_dependencies.py

@@ -2,6 +2,7 @@ import pathlib
 from dataclasses import dataclass
 
 import pytest
+
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.either import is_successful
 from codeflash.models.models import FunctionParent

tests/test_get_read_writable_code.py

@@ -1,6 +1,7 @@
 from textwrap import dedent
 
 import pytest
+
 from codeflash.context.code_context_extractor import get_read_writable_code
 
 

tests/test_instrumentation_run_results_aiservice.py

@@ -6,6 +6,7 @@ from argparse import Namespace
 from pathlib import Path
 
 import isort
+
 from code_to_optimize.bubble_sort_method import BubbleSorter
 from codeflash.code_utils.code_utils import get_run_tmp_file
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize

tests/test_remove_functions_from_generated_tests.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 import pytest
+
 from codeflash.code_utils.remove_generated_tests import remove_functions_from_generated_tests
 from codeflash.models.models import GeneratedTests, GeneratedTestsList
 

tests/test_validate_python_code.py

@@ -1,7 +1,9 @@
-from codeflash.models.models import CodeString
 import pytest
 from pydantic import ValidationError
 
+from codeflash.models.models import CodeString
+
+
 def test_python_string():
     code = CodeString(code="print('Hello, World!')")
     assert code.code == "print('Hello, World!')"
@@ -38,4 +40,4 @@ def test_whitespace_only():
     # Whitespace is still syntactically valid (no-op)
     whitespace_code = "    "
     cs = CodeString(code=whitespace_code)
-    assert cs.code == whitespace_code
+    assert cs.code == whitespace_code