,ji#2SSKrSr\"SS\55r\"SS\55r\"SS\55r\"S S \55r\"S S \55r\"S S\55r\"SS\55r \"SS\55r \"SS\55r g)NcSUlU$)Nz einx.errors) __module__)objs ڧC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\einx/_src/frontend/errors.pyexportrs"CN Jc\rSrSrSrg) EinxError N)__name__r __qualname____firstlineno____static_attributes__r rrr r srr c"\rSrSrSrSSjrSrg) SyntaxErrora 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" ^ Nc^[T[5(d [S5eSSKJn U"T5n[U[ 5(a U/UlOUc/UlO[U5Ul[U4SjUR 55(deSU;a+URSURUR 55n[RX5 g)Nzexpression must be a stringr)ExpressionIndicatorc3V># UHoS:=(a U[T5:v M g7frN)len).0p expressions r 'SyntaxError.__init__..)s$D8a61a#j/118s&)%EXPR%) isinstancestr ValueErroreinx._src.namedtensorrintposlistallreplacecreater __init__)selfrmessager$r indicators ` rr)SyntaxError.__init__s*c**:; ;=' 3 c3  uDH [DHCyDHD488DDDDD w ooh 0@0@0JKG4)rr$Nr rrr__doc__r)rr rrrrs  *rrc"\rSrSrSrSSjrSrg) RankError1a 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^Uc/n[U[5(a U/UlOUc/UlO[U5Ul[ U4SjUR55(de/nUHhn[U[ 5(aUR U5 M+[U5S:XdeUScMDUScMLUR USSUS35 Mj SU;a5URSTRRUR55n[U5S:aUS- nUH nUSU3- nM US TR5-- n[RX5 g) Nc3j># UH(oS:=(a U[TR5:v M* g7frrrrr invocations rr%RankError.__init__..I+Oh6 The following equation(s) were determined for the expression:  rr#r$r%r&r appendrr'r,r(to_call_signature_stringr r)r*r9r+r$ constraints constraints2 constraints ` rr)RankError.__init__?sN  K c3  uDH [DHCyDHOdhhOOOOO %J*c**##J/:!+++a=,A1J '':a=/Z]O(LM & w ooh 0D0D0K0KDHH0UVG | q X XG* VJ<00+4*==???4)rr.NNr0r rrr3r31s  *rr3c"\rSrSrSrSSjrSrg) AxisSizeError_aS 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) Nc^Uc/n[U[5(a U/UlOUc/UlO[U5Ul[ U4SjUR55(de/nUHhn[U[ 5(aUR U5 M+[U5S:XdeUScMDUScMLUR USSUS35 Mj SU;a5URSTRRUR55n[U5S:aUS- nUH nUSU3- nM US TR5-- n[RX5 g) Nc3j># UH(oS:=(a U[TR5:v M* g7frr7r8s rr)AxisSizeError.__init__..{r;r<r=rr>r?rzT The expression, tensor shapes and contraints resulted in the following equation(s):r@rArBrEs ` rr)AxisSizeError.__init__qsN  K c3  uDH [DHCyDHOdhhOOOOO %J*c**##J/:!+++a=,A1J '':a=/Z]O(LM & w ooh 0D0D0K0KDHH0UVG | q n nG* VJ<00+4*==???4)rr.rJr0r rrrLrL_s  *rrLc"\rSrSrSrSSjrSrg) SemanticErrora# 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) Nc^[U[5(deTb[U[5(a U/UlOUc/UlO[ U5Ul[ U4SjUR55(deSU;a5UR STRRUR55nUSTR5-- nOSUl[RX5 g)Nc3j># UH(oS:=(a U[TR5:v M* g7frr7r8s rr)SemanticError.__init__..s+S(QAv@!c**?*?&@"@@(r<rrA) rr r#r$r%r&r'r,r(rDr r))r*r+r9r$s ` rr)SemanticError.__init__s'3''''  !#s##59S$((SSS SS7"!//(J4H4H4O4OPTPXPX4YZ tjAACC CGDH4)rr.rJr0r rrrSrSs  *rrSc"\rSrSrSrSSjrSrg)OperationNotSupportedErrora 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. Nc8UcSn[RX5 g)Nz0This operation is not supported by this backend.)r r))r*r+s rr)#OperationNotSupportedError.__init__s ?HG4)rr r/r0r rrrZrZs *rrZc\rSrSrSrSrg)ImportBackendErrorar 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: ... r Nr rrrr1rr rrr_r_  rr_c\rSrSrSrSrg)BackendResolutionErrora9 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: ... r Nrar rrrdrdrbrrdc(\rSrSrSr\S5rSrg)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 c:SnUbUS- n[UR5S5Hup4X#SSUS3- nM URbo[R"UR5nU(aHUVs/sH ofR S:XdMUR PM" nn[U5S:Xa USUS S 3- nUS - n[[U5S 5(aU[U5R- n[[U55S :aUSU3- n[U5$s snf) NzDThe function that was compiled for this operation failed to execute.z# The following code was generated: r>4z: rAzzThe error occurred at line rz. z The error was: r ) enumerate splitlines __traceback__ traceback extract_tbfilenamelinenorhasattrtyper r rg)ecoder+ilinetbframelinenoss rr(CallOperationError.createsX   = =G$T__%6:U"TF"--;*))!//:9;\~~Q[?[|u||G\7|q(%@ C#PP&& 47J ' ' $q'**+ ,G s1v;? A3x G!'**]s 1DDr N)r rrrr1 staticmethodr(rr rrrgrgs ++rrg) rnr Exceptionr rr3rLrSrZr_rdrgr rrr~s    *)**D** ****Z.*I.*.*b"*I"*"*J * * *         Y    "+"+"+r