tff.utils.update_state

View source on GitHub

Returns a new state with new values for fields in kwargs.

tff.utils.update_state(
    state, **kwargs
)

Args:

• state: the structure with named fields to update.
• **kwargs: the list of key-value pairs of fields to update in state.

Raises:

• KeyError: if kwargs contains a field that is not in state.
• TypeError: if state is not a structure with named fields.