Update functions_to_optimize.py

Author: KRRT7

codeflash/discovery/functions_to_optimize.py

@@ -458,24 +458,21 @@ 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:
-                    console.print("this path contains blocklisted functions")
-                    if function.function_name in blocklist_funcs[path]:
-                        console.print(
-                            f"blocklist: Removing {function.function_name} in {path} because it is blocklisted"
-                        )
-                        functions.remove(function)
-                    else:
-                        console.print(
-                            f"blocklist: Keeping {function.function_name} in {path} because it is not blocklisted"
-                        )
-                else:
+            filtered_functions = []
+            for function in functions:
+                should_keep = True
+
+                rel_path = str(Path(function.file_path).relative_to(project_root))
+
+                if rel_path in blocklist_funcs and function.function_name in blocklist_funcs[rel_path]:
                     console.print(
-                        f"blocklist: Keeping {function.function_name} in {path} because the path is not in the blocklist"
+                        f"blocklist: Removing {function.function_name} in {rel_path} because it is blocklisted"
                     )
+                    should_keep = False
 
+                if should_keep:
+                    filtered_functions.append(function)
+            functions = filtered_functions
         filtered_modified_functions[file_path] = functions
         functions_count += len(functions)
     if not disable_logs: