einx.errors.CallOperationError

exception einx.errors.CallOperationError

This error occurs if the Python function that einx compiles for an einx operation fails to execute.

Example

>>> op = lambda x: None  # Create some invalid operation
>>> einop = einx.jax.adapt_with_vmap(op)
>>> x = np.random.rand(3, 4)
>>> einop("a [b] -> a", x)
einx.errors.CallOperationError: The function that was compiled for this operation failed to execute.
...
The error was: AssertionError: Expected 1st return value of the adapted function to be a tensor