precommit fix

Author: aseembits93

codeflash/api/aiservice.py

@@ -17,7 +17,6 @@
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.code_utils.git_utils import get_last_commit_author_if_pr_exists, get_repo_owner_and_name
 from codeflash.code_utils.time_utils import humanize_runtime
-from codeflash.github.PrComment import FileDiffContent
 from codeflash.lsp.helpers import is_LSP_enabled
 from codeflash.models.ExperimentMetadata import ExperimentMetadata
 from codeflash.models.models import AIServiceRefinerRequest, CodeStringsMarkdown, OptimizedCandidate
@@ -554,7 +553,18 @@ def get_optimization_impact(
         - 'high','medium' or 'low' optimization impact
 
         """
-        diff_str = '\n'.join([unified_diff_strings(code1=original_code[p], code2=new_code[p], fromfile=Path(p).relative_to(root_dir).as_posix(), tofile=Path(p).relative_to(root_dir).as_posix()) for p in original_code if not is_zero_diff(original_code[p], new_code[p])])
+        diff_str = "\n".join(
+            [
+                unified_diff_strings(
+                    code1=original_code[p],
+                    code2=new_code[p],
+                    fromfile=Path(p).relative_to(root_dir).as_posix(),
+                    tofile=Path(p).relative_to(root_dir).as_posix(),
+                )
+                for p in original_code
+                if not is_zero_diff(original_code[p], new_code[p])
+            ]
+        )
         code_diff = f"```diff\n{diff_str}\n```"
         logger.info("!lsp|Computing Optimization Impact…")
         payload = {
@@ -565,7 +575,7 @@ def get_optimization_impact(
             "coverage_message": coverage_message,
             "replay_tests": replay_tests,
             "concolic_tests": concolic_tests,
-            "speedup": f"{100 + 100*float(explanation.speedup):.2f}%",
+            "speedup": f"{100 + 100 * float(explanation.speedup):.2f}%",
             "loop_count": explanation.winning_benchmarking_test_results.number_of_loops(),
             "benchmark_details": explanation.benchmark_details if explanation.benchmark_details else None,
             "optimized_runtime": humanize_runtime(explanation.best_runtime_ns),

codeflash/optimization/function_optimizer.py

@@ -1411,10 +1411,8 @@ def process_review(
         if raise_pr or staging_review:
             data["root_dir"] = git_root_dir()
             # modify argument of staging vs pr based on the impact
-            opt_impact_response = self.aiservice_client.get_optimization_impact(
-                **data,
-            )
-            if opt_impact_response=='low':
+            opt_impact_response = self.aiservice_client.get_optimization_impact(**data)
+            if opt_impact_response == "low":
                 raise_pr = False
                 staging_review = True