handle no repo error

Author: mohammedahmed18

codeflash/optimization/optimizer.py

@@ -91,19 +91,9 @@ def run(self) -> None:
         if not env_utils.check_formatter_installed(self.args.formatter_cmds):
             return
 
-        repo = git.Repo(search_parent_directories=True)
-        owner, repo_name = get_repo_owner_and_name(repo)
-
-        pr_number = get_pr_number()
-        if pr_number is not None:
-            pr_info = cfapi.get_pr_info(owner, repo_name, pr_number)
-            if pr_info is None:
-                logger.warning(f"Could not find {owner}/{repo}#{pr_number}.")
-                return
-            is_draft = pr_info["draft"]
-            if is_draft:
-                logger.warning("PR is in draft mode, skipping optimization")
-                return
+        if is_pr_draft():
+            logger.warning("PR is in draft mode, skipping optimization")
+            return
 
         function_optimizer = None
         file_to_funcs_to_optimize: dict[Path, list[FunctionToOptimize]]
@@ -320,3 +310,21 @@ def run_with_args(args: Namespace) -> None:
             optimizer.cleanup_temporary_paths()
 
         raise SystemExit from None
+
+
+def is_pr_draft() -> bool:
+    try:
+        repo = git.Repo(search_parent_directories=True)
+        owner, repo_name = get_repo_owner_and_name(repo)
+
+        pr_number = get_pr_number()
+        if pr_number is not None:
+            pr_info = cfapi.get_pr_info(owner, repo_name, pr_number)
+            if pr_info is None:
+                logger.warning(f"Could not find {owner}/{repo}#{pr_number}.")
+                return False
+            is_draft = pr_info["draft"]
+            if is_draft:
+                return True
+    except git.exc.InvalidGitRepositoryError:
+        return False