apply pre-commit lint fixes

Author: HeshamHM28

codeflash/api/aiservice.py

@@ -107,7 +107,7 @@ def optimize_python_code(  # noqa: D417
             "python_version": platform.python_version(),
             "experiment_metadata": experiment_metadata,
             "codeflash_version": codeflash_version,
-            "current_username" : get_last_commit_author_if_pr_exists(None),
+            "current_username": get_last_commit_author_if_pr_exists(None),
             "repo_owner": git_repo_owner,
             "repo_name": git_repo_name,
         }

codeflash/api/cfapi.py

@@ -225,17 +225,13 @@ def add_code_context_hash(code_context_hash: str) -> None:
             {"owner": owner, "repo": repo, "pr_number": pr_number, "code_hash": code_context_hash},
         )
 
-def mark_optimization_success(trace_id: str, is_optimization_found: bool) -> Response:
+
+def mark_optimization_success(trace_id: str, *, is_optimization_found: bool) -> Response:
     """Mark an optimization event as success or not.
 
     :param trace_id: The unique identifier for the optimization event.
     :param is_optimization_found: Boolean indicating whether the optimization was found.
     :return: The response object from the API.
     """
-    payload = {
-        "trace_id": trace_id,
-        "is_optimization_found": is_optimization_found,
-    }
-    return make_cfapi_request(
-        endpoint="/mark-as-success", method="POST", payload=payload
-    )
+    payload = {"trace_id": trace_id, "is_optimization_found": is_optimization_found}
+    return make_cfapi_request(endpoint="/mark-as-success", method="POST", payload=payload)

codeflash/code_utils/git_utils.py

@@ -180,17 +180,17 @@ def remove_git_worktrees(worktree_root: Path | None, worktrees: list[Path]) -> N
 
 
 def get_last_commit_author_if_pr_exists(repo: Repo | None = None) -> str | None:
-    """
-    Return the author's name of the last commit in the current branch if PR_NUMBER is set.
+    """Return the author's name of the last commit in the current branch if PR_NUMBER is set.
+
     Otherwise, return None.
     """
+    if "PR_NUMBER" not in os.environ:
+        return None
     try:
-        if "PR_NUMBER" not in os.environ:
-            return None
-
         repository: Repo = repo if repo else git.Repo(search_parent_directories=True)
         last_commit = repository.head.commit
-        return last_commit.author.name
-    except Exception as e:
+    except Exception:
         logger.exception("Failed to get last commit author.")
         return None
+    else:
+        return last_commit.author.name

codeflash/optimization/function_optimizer.py

@@ -390,8 +390,7 @@ def optimize_function(self) -> Result[BestOptimization, str]:  # noqa: PLR0911
                 else:
                     # Mark optimization success since no PR will be created
                     mark_optimization_success(
-                        trace_id=self.function_trace_id,
-                        is_optimization_found=best_optimization is not None
+                        trace_id=self.function_trace_id, is_optimization_found=best_optimization is not None
                     )
                 self.log_successful_optimization(explanation, generated_tests, exp_type)