a HSick%@sFddlZddlZddlZddlZddlZddlZGdddejZdS)Nc@sPeZdZdZejjddddZej e e dddZ d d Z d d Zd dZdS)AttributeTypeIsSupportedCheckera Checks the ``__init__`` method of a given ``nn.Module`` to ensure that all instance-level attributes can be properly initialized. Specifically, we do type inference based on attribute values...even if the attribute in question has already been typed using Python3-style annotations or ``torch.jit.annotate``. This means that setting an instance-level attribute to ``[]`` (for ``List``), ``{}`` for ``Dict``), or ``None`` (for ``Optional``) isn't enough information for us to properly initialize that attribute. An object of this class can walk a given ``nn.Module``'s AST and determine if it meets our requirements or not. Known limitations 1. We can only check the AST nodes for certain constructs; we can't ``eval`` arbitrary expressions. This means that function calls, class instantiations, and complex expressions that resolve to one of the "empty" values specified above will NOT be flagged as problematic. 2. We match on string literals, so if the user decides to use a non-standard import (e.g. `from typing import List as foo`), we won't catch it. Example: .. code-block:: python class M(torch.nn.Module): def fn(self): return [] def __init__(self): super().__init__() self.x: List[int] = [] def forward(self, x: List[int]): self.x = x return 1 The above code will pass the ``AttributeTypeIsSupportedChecker`` check since we have a function call in ``__init__``. However, it will still fail later with the ``RuntimeError`` "Tried to set nonexistent attribute: x. Did you forget to initialize it in __init__()?". Args: nn_module - The instance of ``torch.nn.Module`` whose ``__init__`` method we wish to check N) nn_modulereturncsttjdk|_t|jj}dddfdd|dD}t t |}t |j|_d|_||dS)N)cSs|}|do|d S)N#z# type:)strip startswith)liner L/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/jit/_check.pyis_useless_commentDszAAttributeTypeIsSupportedChecker.check..is_useless_comment csg|]}|s|qSr r ).0lr r r Gz9AttributeTypeIsSupportedChecker.check..F)sys version_infousing_deprecated_astinspect getsource __class____init__joinsplitastparsetextwrapdedentlist__annotations__keysclass_level_annotationsvisiting_class_level_annvisit)selfr source_linesZinit_astr rr check=s z%AttributeTypeIsSupportedChecker.check)nodeann_typercCs|dkr$t|tjsdS|jrdSnb|dkrHt|tjs|dkr|jsft|tjsfdS|jr|t|tjs|dS|j rdSdS)NListFDictOptionalT) isinstancerr,eltsr-r#rConstant NameConstantvalue)r'r*r+r r r _is_empty_containerTs,    z3AttributeTypeIsSupportedChecker._is_empty_containercCsTz*t|jtjr(|jdj|jvr(d|_Wnty>YdS0| |d|_dS)a  If we're visiting a Call Node (the right-hand side of an assignment statement), we won't be able to check the variable that we're assigning to (the left-hand side of an assignment). Because of this, we need to store this state in visitAssign. (Luckily, we only have to do this if we're assigning to a Call Node, i.e. ``torch.jit.annotate``. If we're using normal Python annotations, we'll be visiting an AnnAssign Node, which has its target built in.) rTNF) r/r3rCalltargetsattrr$r%AttributeError generic_visit)r'r*r r r visit_Assignrs    z,AttributeTypeIsSupportedChecker.visit_AssigncCsz|jjjdkrWdSWnty,YdS0|jj|jvr@dShd}z|jjj|vr^WdSWntytYdS0|jjj}||j|sdSt ddS)z Visit an AnnAssign node in an ``nn.Module``'s ``__init__`` method and see if it conforms to our attribute annotation rules. r'N>r.r,r-The TorchScript type system doesn't support instance-level annotations on empty non-base types in `__init__`. Instead, either 1) use a type annotation in the class body, or 2) wrap the type in `torch.jit.Attribute`.) targetr3idr8r7r$ annotationr4warningswarnr'r*Z containersr+r r r visit_AnnAssigns"     z/AttributeTypeIsSupportedChecker.visit_AnnAssigncCs|jr dSzb|jjjjdks6|jjjdks6|jjdkrB||n(|jjjjdks`|jjjdkrj||Wnty||Yn0t|jdkrdSt |jdt j sdShd}z|jdjj}WntyYdS0||vrdS| |jd|sdSt d dS) z Visit a Call node in an ``nn.Module``'s ``__init__`` method and determine if it's ``torch.jit.annotate``. If so, see if it conforms to our attribute annotation rules. Ntorchjitannotater>r.r,r-r;)r%funcr3r=r7r9r8lenargsr/r Subscriptr4r?r@rAr r r visit_Calls:      z*AttributeTypeIsSupportedChecker.visit_Call)__name__ __module__ __qualname____doc__rCnnModuler)rASTstrboolr4r:rBrLr r r r r s 35r)rrrrrCr? NodeVisitorrr r r r s