Skip to content

Commit 95514f4

Browse files
github-actions[bot]Kevin Turcios
andcommitted
style: auto-fix ruff formatting in function_optimizer.py
Co-authored-by: Kevin Turcios <undefined@users.noreply.github.com>
1 parent f9e7f2a commit 95514f4

1 file changed

Lines changed: 3 additions & 11 deletions

File tree

codeflash/optimization/function_optimizer.py

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -682,11 +682,7 @@ def generate_and_instrument_tests(
682682
test_ext = self.language_support.get_test_file_suffix()
683683
# Show the raw LLM output when available, otherwise the post-processed source
684684
display_source = generated_test.raw_generated_test_source or generated_test.generated_original_test_source
685-
code_print(
686-
display_source,
687-
file_name=f"test_{i + 1}{test_ext}",
688-
language=self.function_to_optimize.language,
689-
)
685+
code_print(display_source, file_name=f"test_{i + 1}{test_ext}", language=self.function_to_optimize.language)
690686
if concolic_test_str:
691687
logger.info(f"Generated test {count_tests}/{count_tests}:")
692688
code_print(concolic_test_str, language=self.function_to_optimize.language)
@@ -1900,9 +1896,7 @@ def setup_and_establish_baseline(
19001896
)
19011897
)
19021898

1903-
def display_repaired_functions(
1904-
self, generated_tests: GeneratedTestsList, reviews: list[TestFileReview]
1905-
) -> None:
1899+
def display_repaired_functions(self, generated_tests: GeneratedTestsList, reviews: list[TestFileReview]) -> None:
19061900
"""Display only the repaired function bodies, not the full test files."""
19071901
import libcst as cst
19081902

@@ -2001,9 +1995,7 @@ def review_and_repair_tests(
20011995
tests_for_review = []
20021996
for i, gt in enumerate(generated_tests.generated_tests):
20031997
failed_fns = failed_by_file.get(gt.behavior_file_path, [])
2004-
failure_details = {
2005-
fn: test_failure_messages[fn] for fn in failed_fns if fn in test_failure_messages
2006-
}
1998+
failure_details = {fn: test_failure_messages[fn] for fn in failed_fns if fn in test_failure_messages}
20071999
tests_for_review.append(
20082000
{
20092001
"test_source": gt.raw_generated_test_source or gt.generated_original_test_source,

0 commit comments

Comments
 (0)