o htE@sddlZddlZddlZddlZddlZddlZddlZddlZddlm 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 j:j;<rddl m=Z=dd l!m>Z>m?Z?dd l@mAZAGd d d ZBGd ddZCddZDddZEddZFddZGddZHddZIddZJddZKdd ZLd!d"ZMd#eNejOfd$d%ZPd&d'ZQd(d)ZRd/d*d+ZSd/d,d-ZTgd.ZUdS)0N)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)OpOverloadPacketc@seZdZddZddZdS)ModulecCs||_||_dSN)namemembers)selfr5r6r8i/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/jit/annotations.py__init__Gs zModule.__init__cCs4z|j|WStytd|jd|dw)NzModule z has no member called )r6KeyError RuntimeErrorr5r7r5r8r8r9 __getattr__Ks  zModule.__getattr__N)__name__ __module__ __qualname__r:r>r8r8r8r9r3Fs r3c @sNeZdZeddejiejeddeieeee e e e d Z ddZdd Zd S) EvalEnvtorchTensortypingr*) rCrDrEr*r(rr)r+rAwaitcCs$||_tjjrt|jd<dSdS)Nr1)rcbrC distributedrpc is_availabler1env)r7rGr8r8r9r:bs zEvalEnv.__init__cCs4||jvr |j|S|jdur||Stt|dSr4)rKrGgetattrbuiltinsr=r8r8r9 __getitem__gs     zEvalEnv.__getitem__N)r?r@rAr3rCrDr*r(rr)r+rrrKr:rNr8r8r8r9rBTs  rBc Cst|tr t|j|}nt||}|dur%|r%|\}}|dd}||f}|durRd\}}ztdt|d}t|}Wn tyGYnw|durRt |||}|S)Nr-)NNr) isinstancer2try_real_annotationsoprjoinr, get_type_line TypeErrorparse_type_line) fnrGloc is_method signature param_types return_type type_linesourcer8r8r9 get_signatureos$       r_cCst|p t|Sr4)inspect isfunctionismethod the_callabler8r8r9is_function_or_methodsrecCs2t|s t|r |j}t|rt|jduSdS)NF)recallable__call__r`getfullargspecvarargsrcr8r8r9 is_varargs rjcCsht|tr|j}t|st|rt|jr|j}t|r+t|r%t|}t |j Sddt |DS)NcSsg|]}t|qSr8)str).0ir8r8r9 z#get_param_names..) rPr2rRrerfrgr#r`unwraprhargsrange)rWn_argsr8r8r9get_param_namess   rtc Csz tdt|d}Wn ttfyYdSw|dur dSt|}t|jdkrEt |jdtj rEt j j |d|jdjdt|jdksUt |jdtjs]t j j |ddS)NrOrr-zCannot instantiate class 'z' in a script functionz$Expected a single top-level function)rrSr,OSErrorrUastparselenbodyrPClassDefrCjitfrontend FrontendErrorr5 FunctionDef)rWrXr^py_astr8r8r9check_fns$   rcCsDt|ddd}t|D]}d|jvrtd|dq t|||S)zLEvaluate statement as long as it does not contain any method/function calls.rOeval)modeCALLz/Type annotation should not contain calls, but 'z' does)compiledisget_instructionsopnamer<r)stmtglobrXbytecodeinsnr8r8r9 _eval_no_calls   rc st|\}}z t|it|}Wnttfy#}ztd|d}~wwt|ts,|f}z t|it|}WnttfyI}ztd|d}~wwfdd|D}|t|fS)zParse a type annotation specified as a comment. Example inputs: # type: (Tensor, torch.Tensor) -> 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 annotationcg|]}t|qSr8) ann_to_typerlannrXr8r9rnz#parse_type_line..) split_type_linerrB NameError SyntaxErrorr<rPtupler) r]rGrX arg_ann_str ret_ann_strarg_anneret_ann arg_typesr8rr9rVs2  rVc s`d|d}tt|}ttfdd|}tdttfdd|}t|dkrXtdttfd d|}t|dkrVtd t|ddd d d dSt|dkrf|dd Sd}g}|D]\}}d|vrz||f}n |vr| |ql|durtdd dd|Ddfdd}t ||} d | } |d d| S)zCTry to find the line containing a comment with the type annotation.# type: cs |dvSNr-r8line type_commentr8r9s zget_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$cs|d Srsearchr) type_patternr8r9rsrz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):cs|dSrrr)wrong_type_patternr8r9r szThe 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: cSsg|]}|dqS)r-r8)rlrr8r8r9rn'roz!get_type_line..zl (See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)cs"||td}|Sr4)findrxstrip)r item_typerr8r9get_parameter_type+sz)get_type_line..get_parameter_typez, z...)splitlist enumeratefilterrerrxr<rkrappendrSmapreplace) r^lines type_lineswrong_type_lines return_lineparameter_type_linesline_numrrtypesparameter_typesr8)rrrr9rTs^            rTcCsTtd}z|d}Wn tytddw|||||ddfS)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)rxindex ValueErrorr<r)r] start_offset arrow_posr8r8r9r5s  $rcszt|Wn tyYdSwjgddjD}tfdd|Dr-dSfddjD}tj}||fS)z8Try to use the Py3.5+ annotation syntax to get the type.NcSsg|]}|jqSr8) annotationrlpr8r8r9rnSsz(try_real_annotations..c3s|]}|juVqdSr4)emptyr)sigr8r9 Vsz'try_real_annotations..csg|]}t|jqSr8)rrrrr8r9rnYs)r`rZrreturn_annotation parametersvaluesallr)rWrX all_annotsrr\r8)rXrr9rQIs   rQrcsZt|}|std|jddd|D}fdd|D}tj|}|s+tS|S)NzNo enum values defined for: ''cSsh|]}t|jqSr8)typevalue)rlvr8r8r9 erz&get_enum_value_type..crr8try_ann_to_type)rltrr8r9rnfrz'get_enum_value_type..)rr __class__rC_Cunify_type_listrget)rrX enum_valuesrir_typesresr8rr9get_enum_value_type`s rc CsPt|tjrdSt|tjtjtjtjtjtjtj tj tj f r&t ddSdS)NTzTorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.F) issubclassrCrD LongTensor DoubleTensor FloatTensor IntTensor ShortTensor HalfTensor CharTensor ByteTensor BoolTensorwarningswarn)rr8r8r9 is_tensorrs( rcCsdSr4r8)inpr8r8r9 _fake_rcbsrcsDt|}|tjjurtS|durtSt |r$t |r$tSt |rCt |dkr8|ddkr8t gSt fdd|DSt|rTt|d}|rTt|St|rt|d}t|d}|durxtd|dd|durtd|ddt||St|rt|dtdr|d}n|d}t|}d} |sJ| t|t|tt|St|r t|ttthkrt Sg} t|D]*} | dur| !tt| } d} | sJ| t|t| t| !| qt"| St#j$j%&rt'|rt(t|dSt)|r,t*t|dSt+|rC|r;t|dnt,} t-| S|turLt.S|turUt/S|tus`|t#j0urdt1S|t2urmt3S|t4urvt5S|t6urt,S|tdurtSt |rt7|d rt8|j9S|t#j:urt;S|t#jurt?S|t#j@urt1St |rt|tAjBrtC|durt#jDjEF|}|G}ntH|}tI|tJ|tK|St |rtC|}|dur|St#jLM|rt#jDjEF|S|durtN}t#jOP||S) Nr-rr8crr8r)rlarr8r9rnrz#try_ann_to_type..Unknown type annotation: '' at zWUnsupported annotation {} could not be resolved because {} could not be resolved. At {}__torch_script_interface__)QrEget_argsr` Signaturerr getInferredrrisclassrr&rxrr$rrr!r highlightr r%rrformatreprrr'setintfloatcomplexrrrrCrHrIrJr0r/r"r r rrr rSymIntrrkrboolrrhasattrr rdevicer GeneratorrStreamrdtypeenumEnumr.r{_script_recursive_compile_classqualified_namerr rr _jit_internalcan_compile_classrr_resolve_type_from_object)rrXrGann_args elem_typekeyr contained valid_typemsginnerr maybe_type elementTypescripted_classr5maybe_script_classr8rr9rs     "  "                 rcCs0t|||}|dur |Std|d|)Nrr)rrr)rrXrGthe_typer8r8r9rs r)rr(rrrr*r&r$rr!r%r'rrr rrrrr rr3r_rrtrVrTrrQrrr4)VrvrMrrr`rrErtextwraprrCtorch._Crrrrrrr r r r r rrrrrrrrrrtorch._jit_internalrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+torch._sourcesr,_stater.rHrIrJr/r0r1 torch._opsr2r3rBr_rerjrtrrrVrTrrQrrrrrrr__all__r8r8r8r9sH \X       C  k