code: def extend_name_stack(name: str) -> Iterator[NameStack]:
      prev_context = _source_info_context.context
      curr_name_stack = prev_context.name_stack
      new_context = prev_context.replace(
          name_stack=curr_name_stack.extend(name)
      )
      _source_info_context.context = new_context
      try:
          yield _source_info_context.context.name_stack
      finally:
          _source_info_context.context = prev_context
  
imports: from __future__ import annotations
  from collections.abc import Iterator
  import contextlib
  import dataclasses
  import functools
  import itertools
  import os.path
  import sys
  import sysconfig
  import threading
  import types
  from typing import NamedTuple
  import jax.version
  from jax._src.lib import xla_client
  from jax._src import traceback_util
label_desc: Use image node_id 14 for calling a global function with example usage: extend_name_stack(name) without return types