Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions codeflash/cli_cmds/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,9 @@ def parse_args() -> Namespace:
parser.add_argument(
"--no-pr", action="store_true", help="Do not create a PR for the optimization, only update the code locally."
)
parser.add_argument(
"--no-gen-test", action="store_true", help="Do not generate tests, use only existing tests for optimization."
)
parser.add_argument("--staging-review", action="store_true", help="Upload optimizations to staging for review")
parser.add_argument(
"--verify-setup",
Expand Down
5 changes: 4 additions & 1 deletion codeflash/optimization/function_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -1344,7 +1344,10 @@ def generate_tests(

logger.info(f"!lsp|Generated '{count_tests}' tests for '{self.function_to_optimize.function_name}'")

generated_tests = GeneratedTestsList(generated_tests=tests)
if self.args.no_gen_test:
generated_tests = GeneratedTestsList(generated_tests=[])
else:
generated_tests = GeneratedTestsList(generated_tests=tests)
return Success((count_tests, generated_tests, function_to_concolic_tests, concolic_test_str))

def generate_optimizations(
Expand Down
Loading