code: def assert_max_traces(
      fn: Callable[_P, _T], *, max_traces: Optional[int]
  ) -> Callable[_P, _T]:
      ...
  
imports: import functools
  import inspect
  import weakref
  from collections.abc import Callable
  from typing import Optional, overload, TypeVar
  from typing_extensions import ParamSpec
  from .._custom_types import sentinel
  from .._eval_shape import filter_eval_shape
  from .._module import field, Module, module_update_wrapper
  from .._pretty_print import tree_pformat
  from .._tree import tree_equal
label_desc: Use image node_id 1 for calling a global function with example usage: assert_max_traces(fn) without return types