o ii#@sddlZddZeGdddeZeGdddeZeGdd d eZeGd d d eZeGd d d eZeGdddeZeGdddeZ eGdddeZ eGdddeZ dS)NcCs d|_|S)Nz einx.errors) __module__)objrM/home/ubuntu/.local/lib/python3.10/site-packages/einx/_src/frontend/errors.pyexportsrc@s eZdZdS) EinxErrorN)__name__r __qualname__rrrrr src@eZdZdZdddZdS) SyntaxErrora5 This error occurs if einx operations encounter invalid syntax in an expression. Example: >>> x = np.random.rand(3, 4) >>> einx.id("a (b -> b a", x) einx.errors.SyntaxError: Found an opening parenthesis that is not closed: Expression: "a (b" ^ Ncstts tdddlm}|}t|tr|g|_n |dur%g|_nt||_tfdd|jDs8Jd|vrF| d| |j}t ||dS)Nzexpression must be a stringr)ExpressionIndicatorc3s$|] }|dko |tkVqdSrN)len.0p expressionrr )s"z'SyntaxError.__init__..%EXPR%) isinstancestr ValueErroreinx._src.namedtensorr intposlistallreplacecreater__init__)selfrmessagerr indicatorrrrr s     zSyntaxError.__init__Nrrr __doc__r rrrrr s r c@r ) RankErrora This error occurs if the ranks (i.e. number of dimensions) of all expressions cannot be determined under the given input tensor shapes and additional constraints. Example: >>> x = np.random.rand(3, 4) >>> einx.id("a b c -> a c b", x) einx.errors.RankError: The number of tensor dimensions and axes in the expression does not match. The operation was called with the following arguments: - Positional argument #1: Tensor with shape (3, 4) Nc(|durg}t|tr|g|_n |durg|_nt||_tfdd|jDs+Jg}|D]/}t|tr<||q/t|dksDJ|ddur^|ddur^||dd|dq/d|vrn|dj |j}t|dkr|d7}|D] }|d |7}qz|d 7}t ||dS) Nc3&|]}|dko|tjkVqdSr rrr invocationrrrI$z%RankError.__init__..r = rz> The following equation(s) were determined for the expression:  rrrrrrappendrrr#rto_call_signature_stringrr r!r,r"r constraints constraints2 constraintrr+rr ?0      zRankError.__init__NNr%rrrrr'1s r'c@r ) AxisSizeErrora This error occurs if the size of all axes in an einx expression cannot be determined under the given input tensor shapes and additional constraints. Example: >>> x = np.random.rand(3, 4) >>> y = np.random.rand(4, 4) >>> einx.add("a b, a c -> a b c", x, y) einx.errors.AxisSizeError: Failed to determine the size of all axes in the expression under the given constraints. Expression: "a b, a c -> a b c" The operation was called with the following arguments: - Positional argument #1: Tensor with shape (3, 4) - Positional argument #2: Tensor with shape (4, 4) Ncr() Nc3r)r r*rr+rrr{r-z)AxisSizeError.__init__..r.rr/r0rzT The expression, tensor shapes and contraints resulted in the following equation(s):r1r2r3r6rr+rr qr:zAxisSizeError.__init__r;r%rrrrr<_sr<c@r ) SemanticErroraS This error occurs if the expression requirements of a particular einx operation are not met. Example: >>> x = np.random.rand(3, 4) >>> y = np.random.rand(4, 5) >>> einx.dot("a [b], [c] d -> a d", x, y) einx.errors.SemanticError: All contracted axes must appear in exactly two input expressions. Expression: "a [b], [c] d -> a d" ^ ^ The operation was called with the following arguments: - Positional argument #1: Tensor with shape (3, 4) - Positional argument #2: Tensor with shape (4, 5) Ncst|tsJdurHt|tr|g|_n |durg|_nt||_tfdd|jDs0Jd|vr?|dj|j}|d 7}nd|_t ||dS)Nc3r)r r*rr+rrrr-z)SemanticError.__init__..rr2) rrrrrrrr#rr5rr )r!r"r,rrr+rr s   zSemanticError.__init__r;r%rrrrr=sr=c@r )OperationNotSupportedErrora2 This error occurs if an einx operation is not supported by the selected backend. Example: >>> x = np.random.rand(3, 4) >>> einx.min("a [b]", x, backend="numpy.einsum") einx.errors.OperationNotSupportedError: min operation is not supported by the numpy.einsum backend. NcCs|durd}t||dS)Nz0This operation is not supported by this backend.)rr )r!r"rrrr sz#OperationNotSupportedError.__init__r$r%rrrrr>s r>c@eZdZdZdS)ImportBackendErrora This error occurs if a requested backend failed to be imported or initialized. Example: >>> sys.modules["jax"] = types.SimpleNamespace() # Create some invalid jax module >>> x = np.random.rand(3, 4) >>> einx.min("a [b]", x, backend="jax") einx.errors.ImportBackendError: Failed to import backend "jax" for module "jax" due to the following error: ... Nrrr r&rrrrr@ r@c@r?)BackendResolutionErroraU This error occurs if the backend with which to execute an einx operation could not be determined. Example: >>> x = jnp.zeros((10, 11)) >>> y = torch.zeros(10, 11) >>> einx.add("a b, a b", x, y) einx.errors.BackendResolutionError: Failed to determine which backend to use for this operation: ... NrArrrrrCrBrCc@seZdZdZeddZdS)CallOperationErrora 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 cCsd}|durE|d7}t|dD]\}}||dd|d7}q|jdurEt|j}|rEdd|D}t|dkrE|d |d d 7}|d 7}tt|d rX|t|j7}tt |d krg|d|7}t |S)NzDThe function that was compiled for this operation failed to execute.z# The following code was generated: r/4z: r2cSsg|] }|jdkr|jqS)z)filenamelineno)rframerrr sz-CallOperationError.create..zThe error occurred at line rz. z The error was: r) enumerate splitlines __traceback__ traceback extract_tbrhasattrtyperrrD)ecoder"ilinetblinenosrrrrs"   zCallOperationError.createN)rrr r& staticmethodrrrrrrDs rD) rMr Exceptionrr r'r<r=r>r@rCrDrrrrs("-1%