jax.experimental.checkify.Error

class jax.experimental.checkify.Error(_pred: 'dict[ErrorEffect, Bool]', _code: 'dict[ErrorEffect, Int]', _metadata: 'dict[Int, PyTreeDef]', _payload: 'dict[ErrorEffect, Payload]')

Parameters:

• _pred (dict[ErrorEffect, bool | Array])

• _code (dict[ErrorEffect, int | Array])

• _metadata (dict[int | Array, PyTreeDef])

• _payload (dict[ErrorEffect, list[ndarray | Array]])

__init__(_pred, _code, _metadata, _payload)

Parameters:

• _pred (dict[ErrorEffect, bool | Array])

• _code (dict[ErrorEffect, int | Array])

• _metadata (dict[int | Array, PyTreeDef])

• _payload (dict[ErrorEffect, list[ndarray | Array]])

Return type:

None

Methods

__init__(_pred, _code, _metadata, _payload)
get()	Returns error message if error happened, None if no error happened.
get_exception()	Returns Python exception if error happened, None if no error happened.
throw()
tree_flatten()
tree_unflatten(metadata, data)