code: def __init__(self, fn, max_traces):
      self.fn = fn
      self.max_traces = max_traces
      self.tag = _Weakrefable()
      _traces[self.tag] = 0
      _seen_args[self.tag] = {}
  
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 to create a new _AssertMaxTraces object from inherited base classes: Module with example: obj = _AssertMaxTraces(fn, max_traces)