o 2h4@s2ddlZddlZddlZddlmZddlmZiZedddgZ GdddZ  d0d d Z d d Z e ej d dddedkrfddlmZddlmZe eddddddZe eeddddZd1ddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/ZdS)2N)backend)dmtreeClassRegistrationflatten unflattenc@seZdZddZddZdS)TypeErrorRemappingcCsdSN)selfr r U/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/tree/dmtree_impl.py __enter__"szTypeErrorRemapping.__enter__cCs|tur t||dS)NF) TypeError ValueErrorwith_traceback)r exc_type exc_value tracebackr r r __exit__%szTypeErrorRemapping.__exit__N)__name__ __module__ __qualname__r rr r r r r!s rcCs0|durdd}|dur|j}t||t|<dS)NcSs|Sr) tree_flattenxr r r 1z$register_tree_node..)tree_unflattenrREGISTERED_CLASSES)cls flatten_funcunflatten_funcr r r register_tree_node+s r!cCs t||Sr)r!)rr r r register_tree_node_class7sr"cCs|t||fSr)valueslistkeysdr r r r>srcCstt||Sr) collections OrderedDictzipmetadatachildrenr r r r?s tensorflow) ListWrapper) _DictWrappercCs|dfSrr rr r r rHrcCs tt|Sr)r/r$r+r r r rIs cs,tt}fdd|D}|||fS)Ncsg|]}|qSr r ).0kr&r r Nz*sorted_keys_and_values..)sortedr$r%)r'r%r#r r&r sorted_keys_and_valuesLs r6cCstddt||DS)NcSsi|]\}}||qSr r )r1keychildr r r Usz..)r0r*r+r r r rTscCst|tvp t|Sr)typerr is_nested structurer r r r;Zr;Tcsdtstddtddfddfdd|r.tj|d d S|S) N`func` must be callable, got of type cSst|tr |jdkr |S|S)N MAP_TO_NONE) isinstancer:r)value new_valuer r r remap_map_to_nonedsz#traverse..remap_map_to_nonecsl|}|dur|tjStt|d}|durdS||}fddt|dD}||d|S)Ncsg|] }tj|ddqS)Ttop_down)rtraverser1r)traverse_top_downr r r3qsz7traverse..traverse_top_down..r)rrArgetr:rr$r)sret registration flat_meta_sflat_s)funcrErJr r rJis    z#traverse..traverse_top_downcsttd}|dur'|}fddt|dD}||d|}n1ts/}n)tt j j rIfddt D}t |}nfddD}t |}|}|durb|S|dS)Ncg|]}|qSr r rItraverse_bottom_upr r r3{r4z8traverse..traverse_bottom_up..rrKcsg|]}|qSr r )r1r7)rMrUr r r3scrSr r rIrTr r r3r4)rrLr:rr$rrr;rBr(abcMappingr5_sequence_like)rMrOrPrNfunc_ret)rRrErU)rMr rUws   z$traverse..traverse_bottom_upTrF)callabler r:rrH)rRr=rGr )rRrErUrJr rH^srHcs4t|s|gSgfddtj|ddS)Ncsbtt|d}|durt||d}tj|ddSt|s/||dur-tj S|SdS)NrTrF) rrLr:r$rrrHr;appendrA)rMrOrQr flattenedr r rs zflatten..flatten_funcTrF)r;rrHr<r r\r rs  c Cstt|d}|dur8||}|d}t|dkr|dnt}t||D] \}}t||f||q(dSt |sF| ||fdSt |t jjrat|D] }t||f|||qQdSt|D] \}}t||f||qedS)Nr)rrLr:rlen itertoolscountr*_recursive_flatten_with_pathrr;r[rBr(rVrWr5 enumerate) pathr=r]rOflat_meta_pathsflatpathsr7rCr r r rcs,    rccCs&t|s d|fgSg}td|||S)Nr )r;rc)r=r]r r r flatten_with_paths   ricststddtfdd}t|dkr#t||dSttjg|RWdS1s:wYdS)Nr?r@cs&t|rdS|}|durtjS|Sr)r;rrA)rMrNrRr r func_traverse_wrappers z,map_structure..func_traverse_wrapperrKr)rZr r:r`rHrr map_structure)rR structuresrkr rjr rls  $rlcGs^t|std|dt|ttj||g|RWdS1s(wYdS)Nr?r@)rZr r:rrmap_structure_up_to)shallow_structurerRrmr r r rns$rnc Cstt|d}t|tjrd}tt|d}t|tjr d}||kr9tdt|dt|d|d|d |dur||}||}t|d}t|d}|d|dksptd|dd |dd|d|d t |t |krtd t |d t |d|d|d t ||D] \}} t || qdSt |st |rtd |d |ddSt|ttjtjfrt|ttjtjfstd t|d|d|dt|} t|} | | kstd| d| d|d|d | D] } t || || qdSt|tjjrtdt|d|d|dt|t|ur=tdt|dt|d|d|d t |t |ks[td t |d t |d|d|d t ||D] \}} t || q`dS)Nz*Custom node type mismatch; expected type: z , got type: z while comparing z and .rrKz%Mismatch custom node data; expected: z, got: zArity mismatch; expected: z1Structures don't have the same nested structure: z, zWExpected an instance of dict, collections.OrderedDict, or collections.defaultdict, got z*Dictionary key mismatch; expected key(s): z, got key(s): z7Encountered unregistered collections.abc.Mapping type: zExpected an instance of , got )rrLr:rBr(r)rrr$r`r*assert_same_structurerr;dict defaultdictr5rVrW) aba_registrationb_registration a_flat_meta b_flat_metaa_flatb_flatsub_asub_ba_keysb_keysr7r r r rrs          rrcCs|tddt|D}tddt|D}||kr.cSrr r rr r r r39r4z&`a` and `b` don't have the same paths.z Paths in `a` missing in `b`: z Paths in `b` missing in `a`: )setri differencer)rurva_pathsb_pathsmsga_diffb_diffr r r assert_same_paths7s  rcst|stdkrdStdtdtfddtj|dd}zt\}}td |d tdtyJ|YSw) NrKrzYIncorrect number of leaves provided by `flat_sequence` for `structure`; expected: 1, got rpcstt|d}|dur%||}tjt|ddd}||d|St|sMzt \}}|dur9tj WS|WSt yLt dt dwdS)NrTrFrKz@Too few leaves provided by `flat_sequence` for `structure`. Got rp)rrLr:rrrHr$rr;nextrA StopIterationrr`)rMrOrPrQrrC flat_sequenceflat_sequence_itr r r r Ts&    z(pack_sequence_as..unflatten_funcTrFzGToo many leaves provided by `flat_sequence` for `structure`; expected: rq)r;r`rrdrrHrr)r=rrNindexrr rr pack_sequence_asFs.   rcCsdd}t||ddS)NcSst|tr t|SdSr)rBr$tuple)instancer r r list_to_tuplesr>z&lists_to_tuples..list_to_tupleFrF)rH)r=rr r r lists_to_tuplesrsrcs:tstddtfdd}t||ddS)Nr?r@csPt|ttfrtdd|Dr|}n t|rdS|}|dur%|StjS)Ncss"|] }t|ttdfVqdSr)rBintr:)r1er r r s z>map_shape_structure..map_shape_func..)rBr$rallr;rrA)rrNrjr r map_shape_funcs z+map_shape_structure..map_shape_funcTrF)rZr r:rH)rRr=rr rjr map_shape_structureys   r)NN)T)r(collections.abcrakeras.src.backend.configrkeras.src.utils.module_utilsrr namedtuplerrr!r"r)+tensorflow.python.trackable.data_structuresr/r0r6r;rHrrcrirlrnrrrrrrr r r r sZ         0  S,