o qoi8 @sddlZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZddlmZddlmZdd lmZmZGd d d eeZd ed efddZdeded e eeffddZdedefdede edfde eefded ef ddZded efddZ de e eefe efd efdd Z!d!e ee"edeffded e e"edefffd"d#Z#d$ededed efd%d&Z$d'ed(e eefd efd)d*Z%ej&d+d,d-d'eded(e eefd.ed/ed ef d0d1Z'dS)2N)Enum)dedent)AnyCallableDictListSequenceTupleUnion) OmegaConfSCMode)is_structured_config)_locate)InstantiationException) ConvertMode TargetConfc@s$eZdZdZdZdZdZdZdZdS)_Keysz,Special keys in configs used by instantiate._target_ _convert_ _recursive__args_ _partial_N) __name__ __module__ __qualname____doc__TARGETCONVERT RECURSIVEARGSPARTIALr!r!]/home/ubuntu/.local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.pyrsrxreturncCs(t|tr d|vSt|rd|vSdS)NrF) isinstancedictr is_dict)r#r!r!r" _is_targets  r( input_argskwargscCsP|tjd}|}t|trt|dkr|}||fStdt|jd|d)Nr!rzUnsupported _args_ type: 'z '. value: '') poprrr%rlenrtyper)r)r* config_args output_argsr!r!r"_extract_pos_args#s  r1r.rargsfull_keyc Csvz)t||\}}|D] }t|r|dq |D] }t|r'|dqWn)tyR}zdt|ddt|}|rI|d|7}t||d}~ww|rz t j |g|Ri|WSty}zdt|ddt|}|r|d|7}t||d}~wwz||i|WSty}zdt|d t|}|r|d|7}t||d}~ww) z(Call target (type) with args and kwargs.Nz)Error in collecting args and kwargs for 'z': full_key: zError in creating partial(z, ...) object:zError in call to target 'z': ) r1r is_config _set_parentvalues Exception_convert_target_to_stringreprr functoolspartial) rrr2r*r3argvemsgr!r!r" _call_target2sV            rBtcCst|r |jd|jS|S)N.)callablerr)rCr!r!r"r:dsr:dcCst|tr,i}|D]\}}|dkrt|d}n t|ttfr%t|}|||<q |St|trJg}|D]}t|ttfrBt|}||q5|SJ)Nr)r%r&itemsr:list_prepare_input_dict_or_listappend)rFreskr?r!r!r"rIks"    rItargetc Cst|tr,zt|}Wn ty+}zd|d}|r"|d|7}t||d}~wwt|sId|dt|jd}|rE|d|7}t||S)z>Resolve target string, type or callable into type or callable.zError locating target 'z;', set env var HYDRA_FULL_ERROR=1 to see chained exception.r5Nz!Expected a callable target, got 'z ' of type 'r+)r%strrr9rrEr.r)rMr3r@rAr!r!r"_resolve_targets    rOconfigcOs|durdSt|tr|jdkrttdt|jdt|ttfr(t |}t |}t |s7t|ttfr@t j |ddid}t |rt|}|jgdgd d |||}|rft ||}t ||tjd}|tjtj}|tjd }t|g|R|||d St |rt|}|jgdgd d |||}t ||tjd}|tjtj}|tjd }|rtd t|g|R|||d Sttdt|jd)a4 :param config: An config object describing what to call and what params to use. In addition to the parameters, the config must contain: _target_ : target class or callable name (str) And may contain: _args_: List-like of positional arguments to pass to the target _recursive_: Construct nested objects as well (bool). True by default. may be overridden via a _recursive_ key in the kwargs _convert_: Conversion strategy none : Passed objects are DictConfig and ListConfig, default partial : Passed objects are converted to dict and list, with the exception of Structured Configs (and their fields). object : Passed objects are converted to dict and list. Structured Configs are converted to instances of the backing dataclass / attr class. all : Passed objects are dicts, lists and primitives without a trace of OmegaConf containers. Structured configs are converted to dicts / lists too. _partial_: If True, return functools.partial wrapped method or object False by default. Configure per target. :param args: Optional positional parameters pass-through :param kwargs: Optional named parameters to override parameters in the config object. Parameters not present in the config objects are being passed as is to the target. IMPORTANT: dataclasses instances in kwargs are interpreted as config and cannot be used as passthrough :return: if _target_ is a class name: the instantiated object if _target_ is a callable: the return value of the call Nz???zn Config has missing value for key `_target_`, cannot instantiate. Config type: z Check that the `_target_` key in your dataclass is properly annotated and overridden. A common problem is forgetting to annotate _target_ as a string : '_target_: str = ...' allow_objectsTflags)rQstructreadonly)TFF)rSr8F) recursiveconvertr=zIThe _partial_ keyword is not compatible with top-level list instantiationz2 Cannot instantiate config of type z. Top level config must be an OmegaConf DictConfig/ListConfig object, a plain dict/list, or a Structured Config class or instance.)r%rrrrr.rr&rHrIr r structuredr'copydeepcopy _set_flagr7 _get_parentmergeresolver,rrrrNONEr instantiate_nodeis_list)rPr2r* config_copyrrrr!r!r" instantiates"          rcnoderWcCsft|r1|tjkrtj|dd}|S|tjkr#tj|dtjd}|S|tjkr1tj|dtj d}|S)NTr^)r^structured_config_mode) r r6rALL to_containerr r DICT_CONFIGOBJECT INSTANTIATE)rdrWr!r!r" _convert_nodes    rlTF)rWrVr=rVr=cs|dus t|r|rdSt|s|St|r?tj|vr%|tjntj|vr1|tjntj|vr=|tjn|}|d}t t s]dt }|rY|d|7}t |t |t sxdt |}|rt|rt|d|7}t |t |rfdd|jddD}tjtjtjfvr|Stj|ddid }|||St|r]thd } t|rt|tj|} i} |d d p|} |D]#} | | vrt|| r| rq|| }rt|d }t|| | <qt| ||| |Stjkstjtjfvr)|jjdt fvr)i}|!D]\} }t|d || <q|Stjiddid }|!D]\} }t|d || <q6|||jj|j_tjkr[t"|S|SJdt |j#)Nz4Instantiation: _recursive_ flag must be a bool, got r5z2Instantiation: _partial_ flag must be a bool, got csg|] }t|dqS)rWrV)r`).0itemrmr!r" <s z$instantiate_node..TrerQrR>rrrrrFrmzUnexpected config type : )$r r6_is_noner'rrrr _get_full_keyr%boolr. TypeErrorra_iter_exrrgrjcreater7setr(rOgetrkeys is_missingr`rlrB _metadata object_typer&rG to_objectr)rdrWrVr=r2r3rArGlst exclude_keysrr* is_partialkeyvalue dict_itemscfgr!rmr"r`s                 r`)(rYr<enumrtextwraprtypingrrrrrr r omegaconfr r omegaconf._utilsr hydra._internal.utilsr hydra.errorsr hydra.typesrrrNrrsr(r1rBr:rIr.rOrcrlr_r`r!r!r!r"sj  $        2& s