code: def __init__(self, n: int, child: Doc):
      assert isinstance(child, Doc), child
      self.n = n
      self.child = child
  
imports: from __future__ import annotations
  import abc
  from collections.abc import Sequence
  import enum
  from functools import partial
  import sys
  from typing import NamedTuple
  from jax._src import config
label_desc: Use image node_id 1 to create a new _NestDoc object from inherited base classes: Doc with example: obj = _NestDoc(n, child)