Update code_context_extractor.py

KRRT7 · KRRT7 · commit 09f370d2709f · 2025-04-01T16:51:30.000-05:00
diff --git a/codeflash/context/code_context_extractor.py b/codeflash/context/code_context_extractor.py
@@ -474,7 +474,7 @@ def parse_code_and_prune_cst(
         raise ValueError(msg)
 
     filtered_node, found_target = prune_func(
-        module, target_functions, helpers_of_helper_functions, remove_docstrings=remove_docstrings
+        module, target_functions, helpers_of_helper_functions
     )
 
     if not found_target:

Original file line number	Diff line number	Diff line change
`@@ -474,7 +474,7 @@ def parse_code_and_prune_cst(`
`474`	`474`	`raise ValueError(msg)`
`475`	`475`
`476`	`476`	`filtered_node, found_target = prune_func(`
`477`		`- module, target_functions, helpers_of_helper_functions, remove_docstrings=remove_docstrings`
	`477`	`+ module, target_functions, helpers_of_helper_functions`
`478`	`478`	`)`
`479`	`479`
`480`	`480`	`if not found_target:`