update_states

class brainstate.graph.update_states(node, state_dict, /, *state_dicts)

Update node in place from one or more state mappings.

Parameters:

• node (Any) – The graph node to update in place.

• state_dict (NestedDict or FlattedDict) – The state mapping to apply.

• *state_dicts (NestedDict or FlattedDict) – Additional mappings; merged with state_dict before applying.

Return type:

None