brainstate.transform.check_error

brainstate.transform.check_error(error)

Re-raise a previously captured Error inside a checkified context.

A re-export of jax.experimental.checkify.check_error(). Use it to propagate an Error returned by one checkify()-wrapped function into the error state of an enclosing one.

Parameters:

error (Error) – The error value to propagate.

See also

checkify, check

Return type:

None