etree.Tree

[[Source]]

TODO(b/202712189): Add Any to the Union Tree can be arbitrary chex.dataclass, jax structure,… so has to be Any

Tree = Union[_T, list['Tree'], tuple['Tree', ...], dict[str, 'Tree']]  # pytype: disable=not-supported-yet