states

class brainstate.graph.states(node, *filters, allowed_hierarchy=(0, 2147483647))

Collect State objects from a graph node as ``FlattedDict``(s).

Shared State objects (reachable via multiple paths) are deduplicated by identity: each unique State appears exactly once, matching the behaviour of treefy_states().

Parameters:

• node (Any) – The graph node.

• *filters (type | str | Callable[[Tuple[Key, ...], Any], bool] | bool | EllipsisType | None | Tuple[Filter, ...] | List[Filter]) – Optional predicates/types to partition the collected states.

• allowed_hierarchy (tuple[int, int]) – (lo, hi) graph-node depth bounds.

Returns:

All states (single filter or none), or one mapping per filter.

Return type:

FlattedDict | Tuple[FlattedDict, ...]