debug

Author: KRRT7

codeflash/discovery/functions_to_optimize.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import ast
+import json
 import os
 import random
 import warnings
@@ -156,9 +157,9 @@ def get_functions_to_optimize(
     project_root: Path,
     module_root: Path,
 ) -> tuple[dict[Path, list[FunctionToOptimize]], int]:
-    assert (
-        sum([bool(optimize_all), bool(replay_test), bool(file)]) <= 1
-    ), "Only one of optimize_all, replay_test, or file should be provided"
+    assert sum([bool(optimize_all), bool(replay_test), bool(file)]) <= 1, (
+        "Only one of optimize_all, replay_test, or file should be provided"
+    )
     functions: dict[str, list[FunctionToOptimize]]
     with warnings.catch_warnings():
         warnings.simplefilter(action="ignore", category=SyntaxWarning)
@@ -457,13 +458,31 @@ def filter_functions(
             malformed_paths_count += 1
             continue
         if blocklist_funcs:
-            for function in functions.copy():
-                path = Path(function.file_path).name
-                if path in blocklist_funcs and function.function_name in blocklist_funcs[path]:
-                    functions.remove(function)
-                    logger.debug(f"Skipping {function.function_name} in {path} as it has already been optimized")
-                    continue
-
+            console.print([blocklist_funcs, f"file_path: {file_path}"])
+            filtered_functions = []
+            for function in functions:
+                should_keep = True
+
+                rel_path = str(Path(function.file_path).relative_to(project_root))
+                file_name = Path(rel_path).name
+                console.print(f"rel_path: {rel_path}, file_name: {file_name}")
+
+                if file_name in blocklist_funcs:
+                    qualified_name = (
+                        f"{function.parents[0].name}.{function.function_name}"
+                        if function.parents
+                        else function.function_name
+                    )
+                    if qualified_name in blocklist_funcs[file_name]:
+                        console.print(f"blocklist: Removing {qualified_name} in {rel_path} because it is blocklisted")
+                        should_keep = False
+                    else:
+                        console.print(
+                            f"blocklist: Keeping {qualified_name} in {rel_path} because it is not blocklisted"
+                        )
+                if should_keep:
+                    filtered_functions.append(function)
+            functions = filtered_functions
         filtered_modified_functions[file_path] = functions
         functions_count += len(functions)
     if not disable_logs: