From e3e989070eeedab972dfb2b636066a0aa19db07b Mon Sep 17 00:00:00 2001 From: bagel897 Date: Thu, 30 Jan 2025 10:37:34 -0800 Subject: [PATCH] Update example --- docs/building-with-codegen/traversing-the-call-graph.mdx | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/building-with-codegen/traversing-the-call-graph.mdx b/docs/building-with-codegen/traversing-the-call-graph.mdx index 43336bb3d..27acc5b9c 100644 --- a/docs/building-with-codegen/traversing-the-call-graph.mdx +++ b/docs/building-with-codegen/traversing-the-call-graph.mdx @@ -47,6 +47,10 @@ def create_call_graph(start_func, end_func, max_depth=5): src_func = parent_func.function_definition src_call = parent_func + # Skip external modules + if isinstance(src_func, ExternalModule): + return + # Traverse all function calls for call in src_func.function_calls: func = call.function_definition