struct_handler

Treescope handlers for pz.Struct.

Classes

PyTreeNodeFieldName

Style for attributes that are pytree node children.

Functions

handle_structs(node, path, subtree_renderer)	Renders a penzai struct or layer.
render_short_struct_summary(the_struct)	Renders a short summary of a struct.
render_struct_constructor(node)	Renders the constructor of a Struct, with an open parenthesis.
struct_attr_style_fn_for_fields(fields)	Builds a function to render attributes of a struct.