Skip to content

Commit b4e9ab7

Browse files
formatting
1 parent bdb11d0 commit b4e9ab7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

codeflash/code_utils/code_extractor.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -332,8 +332,9 @@ def add_needed_imports_from_module(
332332
for mod, obj_seq in gatherer.object_mapping.items():
333333
for obj in obj_seq:
334334
if (
335-
f"{mod}.{obj}" in helper_functions_fqn or dst_context.full_module_name is mod
336-
): # avoid circular imports
335+
f"{mod}.{obj}" in helper_functions_fqn
336+
or dst_context.full_module_name is mod # avoid circular imports
337+
):
337338
continue # Skip adding imports for helper functions already in the context
338339
AddImportsVisitor.add_needed_import(dst_context, mod, obj)
339340
RemoveImportsVisitor.remove_unused_import(dst_context, mod, obj)

0 commit comments

Comments
 (0)