From 4fddb1e10566974135712d905185648f2ce4bad1 Mon Sep 17 00:00:00 2001 From: aseembits93 Date: Mon, 12 May 2025 16:13:26 -0700 Subject: [PATCH 1/2] moving things around --- codeflash/main.py | 3 +++ codeflash/optimization/optimizer.py | 5 ++--- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/codeflash/main.py b/codeflash/main.py index 2ec4c614d..45d28a9eb 100644 --- a/codeflash/main.py +++ b/codeflash/main.py @@ -7,6 +7,7 @@ from codeflash.cli_cmds.cli import parse_args, process_pyproject_config from codeflash.cli_cmds.cmd_init import CODEFLASH_LOGO, ask_run_end_to_end_test from codeflash.cli_cmds.console import paneled_text +from codeflash.code_utils.checkpoint import ask_should_use_checkpoint_get_functions from codeflash.code_utils.config_parser import parse_config_file from codeflash.optimization import optimizer from codeflash.telemetry import posthog_cf @@ -19,6 +20,7 @@ def main() -> None: CODEFLASH_LOGO, panel_args={"title": "https://codeflash.ai", "expand": False}, text_args={"style": "bold gold3"} ) args = parse_args() + if args.command: if args.config_file and Path.exists(args.config_file): pyproject_config, _ = parse_config_file(args.config_file) @@ -35,6 +37,7 @@ def main() -> None: ask_run_end_to_end_test(args) else: args = process_pyproject_config(args) + args.previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(args) init_sentry(not args.disable_telemetry, exclude_errors=True) posthog_cf.initialize_posthog(not args.disable_telemetry) optimizer.run_with_args(args) diff --git a/codeflash/optimization/optimizer.py b/codeflash/optimization/optimizer.py index 10d21def5..de2cc1740 100644 --- a/codeflash/optimization/optimizer.py +++ b/codeflash/optimization/optimizer.py @@ -16,7 +16,7 @@ from codeflash.benchmarking.utils import print_benchmark_table, validate_and_format_benchmark_table from codeflash.cli_cmds.console import console, logger, progress_bar from codeflash.code_utils import env_utils -from codeflash.code_utils.checkpoint import CodeflashRunCheckpoint, ask_should_use_checkpoint_get_functions +from codeflash.code_utils.checkpoint import CodeflashRunCheckpoint from codeflash.code_utils.code_replacer import normalize_code, normalize_node from codeflash.code_utils.code_utils import cleanup_paths, get_run_tmp_file from codeflash.code_utils.static_analysis import analyze_imported_modules, get_first_top_level_function_or_method_ast @@ -85,7 +85,6 @@ def run(self) -> None: function_optimizer = None file_to_funcs_to_optimize: dict[Path, list[FunctionToOptimize]] num_optimizable_functions: int - previous_checkpoint_functions = ask_should_use_checkpoint_get_functions(self.args) # discover functions (file_to_funcs_to_optimize, num_optimizable_functions) = get_functions_to_optimize( optimize_all=self.args.all, @@ -96,7 +95,7 @@ def run(self) -> None: ignore_paths=self.args.ignore_paths, project_root=self.args.project_root, module_root=self.args.module_root, - previous_checkpoint_functions=previous_checkpoint_functions, + previous_checkpoint_functions=self.args.previous_checkpoint_functions, ) function_benchmark_timings: dict[str, dict[BenchmarkKey, int]] = {} total_benchmark_timings: dict[BenchmarkKey, int] = {} From d1471b6979257c0e0d3371bb16b5e2a09f72e74e Mon Sep 17 00:00:00 2001 From: Aseem Saxena Date: Mon, 12 May 2025 16:15:44 -0700 Subject: [PATCH 2/2] Update codeflash/main.py --- codeflash/main.py | 1 - 1 file changed, 1 deletion(-) diff --git a/codeflash/main.py b/codeflash/main.py index 45d28a9eb..02b13d5aa 100644 --- a/codeflash/main.py +++ b/codeflash/main.py @@ -20,7 +20,6 @@ def main() -> None: CODEFLASH_LOGO, panel_args={"title": "https://codeflash.ai", "expand": False}, text_args={"style": "bold gold3"} ) args = parse_args() - if args.command: if args.config_file and Path.exists(args.config_file): pyproject_config, _ = parse_config_file(args.config_file)