Update example to exclude external modules

[bagel897]

Motivation

Content

Testing

Please check the following before marking your PR as ready for review

• I have added tests for my changes
• I have updated the documentation or added new documentation as needed
• I have read and agree to the Contributor License Agreement

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

✅ All tests successful. No failed tests found.

Additional details and impacted files

[JohnPeng47]

Thanks. Im good to close this out, but just an FYI, I needed this line or else the call fails as well:

    def traverse_calls(parent_func, current_depth):
        if current_depth > max_depth:
            return

        # Determine source node
        if isinstance(parent_func, Function):
            src_call = src_func = parent_func
        else:
            src_func = parent_func.function_definition
            src_call = parent_func


        print(f"Func[{current_depth}]: {src_func.name}")
        
        # Missing definitions for python libs
        if isinstance(src_func, ExternalModule):
            return

        # Traverse all function calls
        for call in src_func.function_calls:
            func = call.function_definition

            # THIS LINE IS ALSO REQUIRED; FAILS IF NOT ADDED
            if not func:
                continue

            # Skip recursive calls
            if func.name == src_func.name:
                continue

            # Add nodes and edges
            G.add_node(call)
            G.add_edge(src_call, call)

            # Check if we reached the target
            if func == end_func:
                G.add_edge(call, end_func)
                return

            # Continue traversal
            traverse_calls(call, current_depth + 1)

[JohnPeng47]

Yeah Im good to close