Skip to content

Commit 1508ee3

Browse files
committed
test wf changes(test PR2)
1 parent 378d88b commit 1508ee3

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

codeflash/context/code_context_extractor.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ def extract_code_string_context_from_files(
137137
helpers_of_fto: dict[Path, set[FunctionSource]],
138138
helpers_of_helpers: dict[Path, set[FunctionSource]],
139139
project_root_path: Path,
140+
*,
140141
remove_docstrings: bool = False,
141142
code_context_type: CodeContextType = CodeContextType.READ_ONLY,
142143
) -> CodeString:
@@ -413,7 +414,12 @@ def get_function_sources_from_jedi(
413414
and not belongs_to_function_qualified(definition, qualified_function_name)
414415
and definition.full_name.startswith(definition.module_name)
415416
# Avoid nested functions or classes. Only class.function is allowed
416-
and len((qualified_name := get_qualified_name(definition.module_name, definition.full_name)).split(".")) <= 2
417+
and len(
418+
(qualified_name := get_qualified_name(definition.module_name, definition.full_name)).split(
419+
"."
420+
)
421+
)
422+
<= 2
417423
):
418424
function_source = FunctionSource(
419425
file_path=definition_path,

0 commit comments

Comments
 (0)