Skip to content

Commit 7afba98

Browse files
committed
formatting
1 parent 1043272 commit 7afba98

2 files changed

Lines changed: 4 additions & 13 deletions

File tree

codeflash/api/aiservice.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@
2121
from codeflash.models.models import (
2222
AIServiceRefinerRequest,
2323
CodeStringsMarkdown,
24+
OptimizationReviewResult,
2425
OptimizedCandidate,
2526
OptimizedCandidateSource,
26-
OptimizationReviewResult,
2727
)
2828
from codeflash.telemetry.posthog_cf import ph
2929
from codeflash.version import __version__ as codeflash_version
@@ -712,8 +712,7 @@ def get_optimization_review(
712712
if response.status_code == 200:
713713
data = response.json()
714714
return OptimizationReviewResult(
715-
review=cast("str", data["review"]),
716-
explanation=cast("str", data.get("review_explanation", "")),
715+
review=cast("str", data["review"]), explanation=cast("str", data.get("review_explanation", ""))
717716
)
718717
try:
719718
error = cast("str", response.json()["error"])

codeflash/optimization/function_optimizer.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1859,9 +1859,7 @@ def process_review(
18591859
"medium": ("[bold yellow]Medium[/bold yellow]", "yellow", "Review recommended before merging"),
18601860
"low": ("[bold red]Low[/bold red]", "red", "Not recommended to merge"),
18611861
}
1862-
display_info = review_display.get(
1863-
opt_review_result.review.lower(), ("[bold]Unknown[/bold]", "white", "")
1864-
)
1862+
display_info = review_display.get(opt_review_result.review.lower(), ("[bold]Unknown[/bold]", "white", ""))
18651863
explanation_text = opt_review_result.explanation.strip() if opt_review_result.explanation else ""
18661864
if is_LSP_enabled():
18671865
md_content = f"### Reviewer Assessment: {opt_review_result.review.capitalize()}\n{display_info[2]}"
@@ -1872,13 +1870,7 @@ def process_review(
18721870
panel_content = f"Reviewer Assessment: {display_info[0]}\n{display_info[2]}"
18731871
if explanation_text:
18741872
panel_content += f"\n\n[dim]{explanation_text}[/dim]"
1875-
console.print(
1876-
Panel(
1877-
panel_content,
1878-
title="Optimization Review",
1879-
border_style=display_info[1],
1880-
)
1881-
)
1873+
console.print(Panel(panel_content, title="Optimization Review", border_style=display_info[1]))
18821874

18831875
if raise_pr or staging_review:
18841876
data["root_dir"] = git_root_dir()

0 commit comments

Comments
 (0)