added a flag arg to skip draft pr optimization

Author: mohammedahmed18

codeflash/cli_cmds/cli.py

@@ -62,6 +62,8 @@ def parse_args() -> Namespace:
         type=str,
         help="Path to the directory of the project, where all the pytest-benchmark tests are located.",
     )
+    parser.add_argument("--no-draft", default=False, action="store_true", help="Skip optimization for draft PRs")
+
     args: Namespace = parser.parse_args()
     return process_and_validate_cmd_args(args)
 

codeflash/optimization/optimizer.py

@@ -88,7 +88,7 @@ def run(self) -> None:
         if not env_utils.check_formatter_installed(self.args.formatter_cmds):
             return
 
-        if is_pr_draft():
+        if self.args.no_draft and is_pr_draft():
             logger.warning("PR is in draft mode, skipping optimization")
             return