formatter

formatter

Author: KRRT7

codeflash/cli_cmds/cli.py

@@ -150,7 +150,7 @@ def process_and_validate_cmd_args(args: Namespace) -> Namespace:
                 exit_with_message(
                     f"Cannot determine repository information: {e}\n"
                     f"Please ensure your git repository has a remote configured, or use '--no-pr' to optimize locally.",
-                    error_on_exit=True
+                    error_on_exit=True,
                 )
     if args.replay_test:
         for test_path in args.replay_test:
@@ -272,15 +272,18 @@ def handle_optimize_all_arg_parsing(args: Namespace) -> Namespace:
                 apologize_and_exit()
             git_remote = getattr(args, "git_remote", None)
             if not check_and_push_branch(git_repo, git_remote=git_remote):
-                exit_with_message("Cannot proceed without a valid git remote configuration. See error message above.", error_on_exit=True)
+                exit_with_message(
+                    "Cannot proceed without a valid git remote configuration. See error message above.",
+                    error_on_exit=True,
+                )
             try:
                 owner, repo = get_repo_owner_and_name(git_repo)
                 require_github_app_or_exit(owner, repo)
             except ValueError as e:
                 exit_with_message(
                     f"Cannot determine repository information: {e}\n"
                     f"Please ensure your git repository has a remote configured, or use '--no-pr' to optimize locally.",
-                    error_on_exit=True
+                    error_on_exit=True,
                 )
     if not hasattr(args, "all"):
         args.all = None

codeflash/code_utils/git_utils.py

@@ -82,9 +82,10 @@ def get_remote_url(repo: Repo | None = None, git_remote: str | None = "origin")
     repository: Repo = repo if repo else git.Repo(search_parent_directories=True)
     available_remotes = get_git_remotes(repository)
     if not available_remotes:
-        raise ValueError(f"No git remotes configured in this repository")
+        raise ValueError("No git remotes configured in this repository")
     if git_remote not in available_remotes:
-        raise ValueError(f"Git remote '{git_remote}' does not exist. Available remotes: {', '.join(available_remotes)}")
+        msg = f"Git remote '{git_remote}' does not exist. Available remotes: {', '.join(available_remotes)}"
+        raise ValueError(msg)
     return repository.remote(name=git_remote).url
 
 
@@ -144,7 +145,7 @@ def check_and_push_branch(repo: git.Repo, git_remote: str | None = "origin", *,
             f"Alternatively, you can run codeflash with the '--no-pr' flag to optimize locally without creating PRs."
         )
         return False
-    
+
     # Check if the specified remote exists
     if git_remote not in available_remotes:
         logger.error(
@@ -156,7 +157,7 @@ def check_and_push_branch(repo: git.Repo, git_remote: str | None = "origin", *,
             f"  3. Run codeflash with '--no-pr' to optimize locally without creating PRs."
         )
         return False
-    
+
     remote = repo.remote(name=git_remote)
 
     # Check if the branch is pushed