Skip to content

Commit 5dfdc79

Browse files
fix type casting
1 parent f2f4ce7 commit 5dfdc79

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

codeflash/optimization/optimizer.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -91,14 +91,16 @@ def run(self) -> None:
9191

9292
owner, repo = get_repo_owner_and_name()
9393

94-
pr_info = cfapi.get_pr_info(owner, repo, int(get_pr_number()))
95-
if pr_info is None:
96-
logger.warning(f"Could not find {owner}/{repo}#{get_pr_number()}.")
97-
return
98-
is_draft = pr_info["draft"]
99-
if is_draft:
100-
logger.warning("PR is in draft mode, skipping optimization")
101-
return
94+
pr_number = get_pr_number()
95+
if pr_number is not None:
96+
pr_info = cfapi.get_pr_info(owner, repo, pr_number)
97+
if pr_info is None:
98+
logger.warning(f"Could not find {owner}/{repo}#{pr_number}.")
99+
return
100+
is_draft = pr_info["draft"]
101+
if is_draft:
102+
logger.warning("PR is in draft mode, skipping optimization")
103+
return
102104

103105
function_optimizer = None
104106
file_to_funcs_to_optimize: dict[Path, list[FunctionToOptimize]]

0 commit comments

Comments
 (0)