HardcodedStructureHandler

class penzai.treescope.handlers.hardcoded_structure_handlers.HardcodedStructureHandler

Bases: object

A handler for a specific hardcoded list of dataclass-like/enum-like types.

Each of these types will be shown like a dataclass or an enum. This is intended to support structures that act like dataclasses, namedtuples, or enums, but are not implemented as such (e.g. JAX’s ShapeDtypeStruct.)

Variables:

known_structure_types (dict[type[Any], HasFieldsInClassAttr | HasExplicitFields | IsEnumLike]) – Mapping from handled types to a tuple of their attribute names.

Methods

__init__(known_structure_types)
__call__(node, path, subtree_renderer)	Renders the hardcoded types from known_structure_types.

Attributes

known_structure_types

__call__(node: Any, path: tuple[Any, ...] | None, subtree_renderer: renderer.TreescopeSubtreeRenderer)

Renders the hardcoded types from known_structure_types.