jaxvacua.util.unflatten_func_class

unflatten_func_class(aux_data, children, myclass)

Inverse of flatten_func() for a specific class myclass.

Bypasses __init__ (which often has side-effects) — restores the object via object.__new__ + setattr of each saved attribute.

Parameters:

• aux_data (Tuple[Any, ...]) – Auxiliary data from flatten_func().

• children (Tuple[Any, ...]) – Children (traced values) from flatten_func().

• myclass (type) – Class to reconstruct.

Returns:

myclass – A fresh instance with all flattened attributes restored.

Return type:

Any