itUddlZddlZddlmZmZmZmZmZddlm Z dZ e e d<GddZ Gdd Zy) N)ListOptionalSetTupleUnion) EinopsErroru… _ellipsisc"eZdZdZdefdZdZy) AnonymousAxiszHImportant thing: all instances of this class are not equal to each othervaluect||_|jdkr2|jdk(r tdtd|jy)NzENo need to create anonymous axis of length 1. Report this as an issuez0Anonymous axis should have positive length, not )intr r)selfr s c/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/einops/parsing.py__init__zAnonymousAxis.__init__ sOZ ::?zzQ!"ijj!$TUYU_U_T`"abb c2t|jdS)Nz-axis)strr )rs r__repr__zAnonymousAxis.__repr__sdjj/"%((rN)__name__ __module__ __qualname____doc__rrrrrr r sRccc)rr c eZdZdZddddededefdZdefd Zdefd Z e dd edede eeffd Z e d edefd Z y)ParsedExpressionz non-mutable structure that contains information about one side of expression (e.g. 'b c (h w)') and keeps some information important for downstream F)allow_underscoreallow_duplicates expressionrrcd_d_t_d_g_d|vrid|vr t dtj|ddk7stj|ddk7r t d|jdt}d_dfd }d}|D]}|d vrR|||d}|d k(r t d g(|d k(s. t dj jdYtj|s |dtfvr ||}}||z }t d|dt d|d| ||yy)NF.z...z6Expression may contain dots only inside ellipsis (...)rzUExpression may contain dots only inside ellipsis (...); only one ellipsis for tensor Tc|jvrr|dk(sstd|d|tk(rejjt'jj td_yj td_ytj|}|r.t|dk(r jj gy yj|\}}|s|std|d ||r t|}jj||rd_ jj |gyj |y) N_z2Indexing expression contains duplicate dimension ""FTr)rzInvalid axis identifier:  ) identifiersrr add compositionappendhas_ellipsis_parenthesizedr isdecimalrcheck_axis_name_return_reasonr has_non_unitary_anonymous_axes)x is_number is_axis_namereasonrr bracket_grouprs r add_axis_namez0ParsedExpression.__init__..add_axis_name3sYD$$$(Q#X?O%(Z[\Z]]^&_``I~  $$Y/ ($$++I66;D3!((36:D3MM!, Q1$,((//3'+'I'I!^n'I'o$ f!\%(A!Bvh&OPP%a(A  $$Q':>D7 ($$++QC0!((+rz() (zDAxis composition is one-level (brackets inside brackets not allowed))zBrackets are not balancedr%zUnknown character ''z'Imbalanced parentheses in expression: "r&) has_ellipsisr,setr(r/r*rrcountreplacer r+isalnum)rr rrr5current_identifiercharr4s` `` @rrzParsedExpression.__init__s"':>'%(U49+8: * J&!"Z[[yyU+q0CIIj#4NRS4S!k$++E9=J $D -1  ,D"Du}%1!"45%)"3;$0)*pqq$&MS[$,)*EFF$$++M:$(MT"dsI.>&>%-)-&&$.&!$7vQ"?@@),  $ G |STUV V  ) , - *rreturncg}|jD]1}t|tsJd|D]}|j|3|S)Nzdoes not work with ellipsis)r* isinstancelistr+)rresult composed_axisaxiss rflat_axes_orderz ParsedExpression.flat_axes_orderqsJ!--MmT2 Q4Q Q2% d#&. rch|jD]#}t|tst|dkDs#yy)NrTF)r*rBrClen)raxess rhas_composed_axesz"ParsedExpression.has_composed_axesys.$$D$%#d)a-%rnamec tj|sy|ddk(s|ddk(r |dk(r|ryytj|rt j d|t d |d vrt j d td y) N)Fznot a valid python identifierrr%)T)Fz8axis name should should not start or end with underscorez7It is discouraged to use axes names that are keywords: ) stacklevel)rFzQIt is discouraged to use 'axis' as an axis name and will raise an error in future)r isidentifierkeyword iskeywordwarningswarnRuntimeWarning FutureWarning)rLrs rr.z.ParsedExpression.check_axis_name_return_reasons%9 !W^tBx3s{/T  & MdVT"  x g!  rc6tj|\}}|S)z Valid axes names are python identifiers except keywords, and additionally should not start or end with underscore )rr.)rLis_valid_reasons rcheck_axis_namez ParsedExpression.check_axis_names -JJ4P'rN)F)rrrrrboolrrrGrK staticmethodrr.r\rrrrrs EJdiP.3P.TP.]aP.d4C4TYZ^`cZcTd,cdrr)rSrUtypingrrrrreinopsrr r__annotations__r rrrrrbs444 3 ) )EEr