i>GLddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7ddl8m9Z9e jtjvjyrddl m=Z=dd l!m>Z>m?Z?dd l@mAZAGd d ZBGd dZCdZDdZEdZFdZGd!dZHdZIdZJdZKdZLdZMdeNejfdZPdeQfdZRd!dZSd"dZTd"dZUgd ZVy)#N)dedent)_GeneratorTypeAnyType AwaitTypeBoolType ComplexType DeviceObjTypeDictTypeEnumType FloatType FutureType InterfaceTypeIntTypeListTypeNoneType NumberType OptionalType StreamObjType StringType TensorType TupleType UnionType)_Await_qualified_nameAnyBroadcastingList1BroadcastingList2BroadcastingList3DictFutureis_awaitis_dict is_future is_ignored_fnis_list is_optionalis_tupleis_unionListOptionalTupleUnion)get_source_lines_and_file)_get_script_class)RRefType)is_rrefRRef)OpOverloadPacketceZdZddZdZy)ModuleNc ||_||_yN)namemembers)selfr8r9s j/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch/jit/annotations.py__init__zModule.__init__Gs  ct |j|S#t$rtd|jd|dwxYw)NzModule z has no member called )r9KeyError RuntimeErrorr8r:r8s r; __getattr__zModule.__getattr__KsJ <<% % $))$:4&A  s&7returnN)__name__ __module__ __qualname__r<rBr=r;r5r5Fs r=r5c eZdZeddej iej eddeieeee e e e d Z d dZdZy) EvalEnvtorchTensortypingr+) rKrLrMr+r)rr*r,r AwaitNc||_tjjj rt |j d<yy)Nr2)rcbrK distributedrpc is_availabler2env)r:rPs r;r<zEvalEnv.__init__bs4    - - /#DHHV  0r=c||jvr|j|S|j|j|Stt|dSr7)rTrPgetattrbuiltinsrAs r; __getitem__zEvalEnv.__getitem__hsC 488 88D> ! 88 88D> !xt,,r=rC)rErFrGr5rKrLr+r)rr*r,r rrTr<rXrHr=r;rJrJTsQ(ELL!9:,,GU#34 C$ -r=rJc>t|trt|j|}n t||}||r|\}}|dd}||f}|Fd\}} t dj t |d}t|}| t|||}|S#t$rYwxYw)Nr.)NNr) isinstancer3try_real_annotationsoprjoinr- get_type_line TypeErrorparse_type_line) fnrPloc is_method signature param_types return_type type_linesources r; get_signaturerjps"&'(4 (S1  $- [!!"o  +. & 6 BGG$=b$A!$DEFF%f-I  ' 3 Tuple[Tensor] # type: (Tensor, Tuple[Tensor, Tensor]) -> Tensor z6Failed to parse the argument list of a type annotationNz4Failed to parse the return type of a type annotation) split_type_linerrJ NameError SyntaxErrorr@r[tuple ann_to_type) rhrPrc arg_ann_str ret_ann_strarg_anneret_annann arg_typess r;raras /y9K R> gu %* R> 3::'3S#&'I: k'3/ //! { # D   { # B   ;s4A7BB=7B BBB:) B55B:c B d |jd}tt|}tt fd|}t j d tt fd|}t |dk(rbt j d tt fd|}t |dkDr&td t|ddzd zd zd zy t |dk(r|ddjSd }g}|D]%\}}d|vr||f}n |vs|j|'|1tddj|Dcgc]}|d c}zdz fd}t||} dj| } |djd| Scc}w)zCTry to find the line containing a comment with the type annotation.# type: c|dvSNr.rH)line type_comments r;zget_type_line..s,$q'*Ar=z!# type:\ ignore(\[[a-zA-Z-]+\])?$c.j|d Srsearch)r type_patterns r;rzget_type_line..sl.A.A$q'.J*Jr=rz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):c,j|dSrr)rwrong_type_patterns r;rzget_type_line.. s 2 9 9$q' Br=zThe annotation prefix in line z* is probably invalid. It must be '# type:'zl See PEP 484 (https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)z for examplesNr.z# type: (...) -> z_Return type line '# type: (...) -> ...' not found on multiline type annotation for type lines: zl (See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)cd||jtzd}|jSr7)findrstrip)r item_typers r;get_parameter_typez)get_type_line..get_parameter_type,s0<03|3DDFG   r=z, z...)splitlist enumeratefilterrerrr@r|rappendr^mapreplace)rilines type_lineswrong_type_lines return_lineparameter_type_linesline_numrrtypesparameter_typesrrrs @@@r;r_r_sL LL E 5! "EfA5IJJ::DELfJJWXJ :!ZZ(TU BE J   1 $0&q)!,-.?@BB# #  ZA !}Q%%''K$$ $ &#T*K  T ! ' ' - %  1iiZ8ZTaZ89 :~ ~  ! "$8 9Eii&O q> ! !% 999s Fctd} |jd}|||j ||dzdj fS#t$r tddwxYw)aSplit the comment with the type annotation into parts for argument and return types. For example, for an input of: # type: (Tensor, torch.Tensor) -> Tuple[Tensor, Tensor] This function will return: ("(Tensor, torch.Tensor)", "Tuple[Tensor, Tensor]") rz->z4Syntax error in type annotation (couldn't find `->`)N)rindex ValueErrorr@r)rh start_offset arrow_poss r;rr6suy>LOOD) \) , 2 2 4i A 6P6V6V6X XX  B  s AAc tj|jgjj Dcgc]}|j c}z}tfd|Dryjj Dcgc]}t|j |}}tj|}||fS#t$rYywxYwcc}wcc}w)z8Try to use the Py3.5+ annotation syntax to get the type.Nc3:K|]}|juywr7)empty).0rsigs r; z'try_real_annotations..Ws 2z3#)) zs) rlrerreturn_annotation parametersvalues annotationallr)rbrcp all_annotsrrgrs @r;r\r\Js#''(!nn335,5 5,J 2z 229<9N9N9PQ9PAQ\\3/9PIQc33S9K k !! , RsCC C C CrcFt|}|std|jd|Dchc]}t|j}}|Dcgc]}t ||}}t jj|}|stjS|Scc}wcc}w)NzNo enum values defined for: '') rr __class__typevaluetry_ann_to_typerK_Cunify_type_listrget)rrc enum_valuesvrtir_typesress r;get_enum_value_typeras#'7K 8 QGHH$/ 0KqT!'']KE 0167A3'H7 (( " "8 ,C {{} J 17s B Bc t|tjryt|tjtjtj tj tjtjtjtjtjf rtjddyy)NTzTorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.r) stacklevelF) issubclassrKrL LongTensor DoubleTensor FloatTensor IntTensor ShortTensor HalfTensor CharTensor ByteTensor BoolTensorwarningswarn)rs r; is_tensorrss#u||$          OO                    H   r=cyr7rH)inps r; _fake_rcbrs r=c  tj|}|tjjurt j S|tjStj|rt|rt jSt|rDt|dk(r|ddk(r tgSt|Dcgc]}t||c}St|rt|d|}|r t!|St#|rrt|d|}t|d|}|"t%d|dd|j'|"t%d|dd|j't)||St+|rkt-|dt/dr|d}n|d}t||} | s/t1dt3|dt3|dt3|t5| St7|rt9|t:t<t>hk(rtAjSg} tj|D]u}|#| jCtjt||} | s/t1dt3|dt3|dt3|| jC| wtE| StFjHjJjMr#tO|rtQt|d|StS|rtUt|d|StW|r0|rt|d|ntYj} t[| S|t<urt]jS|t>urt_jS|t:us|tFj`urtcjS|tdurtgjS|thurtkjS|tlurtYjS|t/durtjStj|r!to|d rtq|jrS|tFjturtwjS|tFjxurt{jS|tFj|urtjS|tFjurtcjS|tFjurtcjStj|rt-|tjrqt|;tFjjj||} | j}n t|}t|t||t|Stj|rXt|}||StFjj|r*tFjjj||S|t}tFjj|||Scc}w) Nr.rrHUnknown type annotation: '' at zUnsupported annotation z could not be resolved because z could not be resolved. At __torch_script_interface__)RrMget_argsrl Signaturerr getInferredrrisclassrr'rrrr%rr"r highlightr r&rrAssertionErrorreprrr(setintfloatcomplexrrrrKrQrRrSr1r0r#r r!rrr rSymIntrr|rboolrrhasattrrrdevicer GeneratorrStreamrdtypeqschemeenumEnumr/r_script_recursive_compile_classqualified_namerr rr _jit_internalcan_compile_classrr_resolve_type_from_object)rrcrPann_argsa elem_typekeyr contained valid_typeinner maybe_type elementTypescripted_classr8maybe_script_classs r;rrsGs#H g%%%%%'' {||~s #~~} x=A (1+"3R= 8D8a/!S18DEEs|#HQK5 I& &s|hqk3/ S1 ;,Xa[Ms}}>OP  =,Xa[Ms}}>OP U##3 hqk4: .  I  I$Y4  )$s)4S ?##?S {L J''} x=S%1 1>># #%Ay X\\^,(C0J$-d3i[8WAwi;DI;H LL $& ))+  S9::~/(1+s;<<};Cohqk37 %% e|}} g~   czSELL({{} cz~~ d{||~ cz{{} d4j||~s-I JS;;<< ell  "" eoo!!## ell  "" ekk{{} emm{{}s 3 : S ! )"YY..GGSQN!002D"3'D1#s;T#YGGs.s3  )% %    0 0 599$$==c3G G { 88 - -c3 <r1s     .,5% %%'!1'  --8:N /, *%0<@:FY(".4 ?$d:r=jO" r=