a þdíã@sdUddlmZddlZddlZddlmZmZmZmZdZ e e d<Gdd„de ƒZ Gdd „d ƒZdS) é)Ú EinopsErrorN)ÚListÚOptionalÚSetÚTupleuâ€¦Ú _ellipsisc@s&eZdZdZedœdd„Zdd„ZdS)Ú AnonymousAxiszIImportant thing: all instances of this class are not equal to each other )ÚvaluecCs<t|ƒ|_|jdkr8|jdkr(tdƒ‚ntd |j¡ƒ‚dS)NézENo need to create anonymous axis of length 1. Report this as an issuez2Anonymous axis should have positive length, not {})Úintr rÚformat)Úselfr ©rúW/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/einops/parsing.pyÚ__init__ s     zAnonymousAxis.__init__cCsd t|jƒ¡S)Nz{}-axis)r Ústrr )r rrrÚ__repr__szAnonymousAxis.__repr__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrr src@steZdZdZddœedœdd„Zedœdd„Zedœd d „Ze de ee ee fd œd d „ƒZ e e edœdd„ƒZ dS)Ú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_underscorecsVdˆ_dˆ_tƒˆ_dˆ_gˆ_d|vrrd|vr8tdƒ‚t |d¡dksXt |d¡dkr`tdƒ‚|  dt ¡}dˆ_d‰‡‡‡fd d „}d}|D]¢}|d vrî||ƒd}|d krĈdur¾td ƒ‚g‰n(|dkrìˆdurÜtdƒ‚ˆj  ˆ¡d‰qŽt  |¡s|dt fvr"|dur|}n||7}qŽtd  |¡ƒ‚qŽˆdurJtd  |¡ƒ‚||ƒdS)NFÚ.z...z6Expression may contain dots only inside ellipsis (...)r ézUExpression may contain dots only inside ellipsis (...); only one ellipsis for tensor Tcs|dur|ˆjvr.ˆr |dks.td |¡ƒ‚|tkrpˆj t¡ˆdur^ˆj t¡dˆ_nˆ t¡dˆ_n¤t  |¡}|r¤t |ƒdkr¤ˆdur ˆj g¡ndSˆj |ˆd\}}|sÎ|sÎtd ||¡ƒ‚|rÚt |ƒ}ˆj |¡|rðdˆ_ ˆdur ˆj |g¡n ˆ |¡dS)NÚ_z5Indexing expression contains duplicate dimension "{}"FTr rzInvalid axis identifier: {} {})Ú identifiersrr rÚaddÚ compositionÚappendÚhas_ellipsis_parenthesizedrÚ isdecimalr Úcheck_axis_name_return_reasonrÚhas_non_unitary_anonymous_axes)ÚxZ is_numberZ is_axis_nameÚreason©rZ bracket_groupr rrÚ add_axis_name0s6         z0ParsedExpression.__init__..add_axis_namez() ú(zDAxis composition is one-level (brackets inside brackets not allowed)ú)zBrackets are not balancedrzUnknown character '{}'z*Imbalanced parentheses in expression: "{}")Z has_ellipsisr!Úsetrr$rrrÚcountÚreplacerr Úisalnumr )r Z expressionrr(Zcurrent_identifierÚcharrr'rrsL ÿ #    zParsedExpression.__init__)ÚreturncCs:g}|jD]*}t|tƒs Jdƒ‚|D]}| |¡q$q |S)Nzdoes not work with ellipsis)rÚ isinstanceÚlistr )r ÚresultZ composed_axisÚaxisrrrÚflat_axes_orderms  z ParsedExpression.flat_axes_ordercCs,|jD] }t|tƒrt|ƒdkrdSqdS)Nr TF)rr1r2Úlen)r ZaxesrrrÚhas_composed_axesus z"ParsedExpression.has_composed_axes)Únamerr0cCsrt |¡sdS|ddks&|ddkr:|dkr6|r6dSdSt |¡rVt d |¡t¡|dvrjt d t¡dSdS) N)Fznot a valid python identifierrréÿÿÿÿ)TÚ)Fz8axis name should should not start or end with underscorez9It is discouraged to use axes names that are keywords: {})r4zQIt is discouraged to use 'axis' as an axis name and will raise an error in future) rÚ isidentifierÚkeywordÚ iskeywordÚwarningsÚwarnr ÚRuntimeWarningÚ FutureWarning)r8rrrrr#|s   ÿz.ParsedExpression.check_axis_name_return_reason)r8r0cCst |¡\}}|S)z‹ Valid axes names are python identifiers except keywords, and additionally should not start or end with underscore )rr#)r8Zis_validZ_reasonrrrÚcheck_axis_nameŒsz ParsedExpression.check_axis_nameN)F)rrrrÚboolrrr5r7Ú staticmethodrrr#rBrrrrrsPr)Zeinopsrr<r>ÚtypingrrrrrrÚ__annotations__ÚobjectrrrrrrÚs