o –hÝã@sdddlmZddlZddlmZmZmZeddƒZdd„Z dd „Z d d „Z d d „Z Gdd„dƒZ dS)é)Ú namedtupleN)ÚShiftÚOpenConstituentÚCloseConstituentÚ RepairEnumzname value is_correctcCsºg}|D]E}|g}|d|…D]}|D] }| ||g¡}qqd} ||D]}| |¡} |j|} | | d| f ¡ ¡7} | ||g¡}q$| | ¡qt |¡} dd„|| Dƒ} || | fS)zÞ score candidate fixed sequences by summing up the transition scores of the most important block the candidate with the best summed score is chosen, and the candidate sequence is reconstructed from the blocks égrcSsg|] }|D]}|‘qqS©r)Ú.0ÚblockÚxrrúd/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/models/constituency/dynamic_oracle.pyÚ sz$score_candidates..)Ú bulk_applyÚforwardÚtransition_mapÚcpuÚitemÚappendÚnpÚargmax)ÚmodelÚstateÚ candidatesÚ candidate_idxÚscoresÚ candidateÚ current_stater Ú transitionÚscoreÚ predictionsÚt_idxÚbest_idxÚbest_candidaterrr Úscore_candidates s"ÿ      r#cCsbd}|t|ƒkr/t||tƒr|d}nt||tƒr%|d}|dkr%|S|d}|t|ƒksdS)z° Advance cur_index through gold_sequence until we have seen 1 more Close than Open The index returned is the index of the Close which occurred after all the stuff rréÿÿÿÿN)ÚlenÚ isinstancerr©Ú gold_sequenceÚ cur_indexÚcountrrr Úadvance_past_constituents!s    úr+cCsbd}|d}|dkr/t||tƒr|d}|dkr|Sn t||tƒr'|d}|d}|dks dS)z† Go backwards from cur_index to find the open which opens the previous block of stuff. Return None if it can't be found. rrN)r&rrr'rrr Úfind_previous_open1sÿùr,cCsŠd}d}|t|ƒkrCt||tƒr|d}n#t||tƒr(|d}|dkr'|Snt||tƒr9|r7|dkr7|Sd}|d}|t|ƒks dS)a Advance cur_index through gold_sequence until the next block has ended This is different from advance_past_constituents in that it will also return when there is a Shift when count == 0. That way, we return the first block of things we know attach to the left rFrr$TN)r%r&rrr)r(r)r*Ú saw_shiftrrr Úfind_in_order_constituent_endCs    õ r.c@seZdZdd„Zdd„ZdS)Ú DynamicOraclecs~||_|dur |nˆjj|_ˆ|_tƒ|_|r&t‡fdd„| d¡Dƒƒ|_tƒ|_|r=t‡fdd„| d¡Dƒƒ|_dSdS)Ncóg|]}ˆ| ¡‘qSr©Úupper©r r ©Ú repair_typesrr r dóz*DynamicOracle.__init__..ú,cr0rr1r3r4rr r gr6) Ú root_labelsÚUNKNOWNÚvalueÚ oracle_levelr5ÚsetÚadditional_levelsÚsplitÚdeactivated_levels)Úselfr8r;r5r=r?rr4r Ú__init__\s"ÿzDynamicOracle.__init__c CsÐ|j|j}||kr|jjdfS|jD]N}|jdurq|jdur/|j|jkr/||jvr/|js/q||j vr5q| |||j|j|j ||¡}|durHqt |t ƒrWt |ƒdkrW|S|dura||fSq|jjdfS)a Return which error has been made, if any, along with an updated transition list We assume the transition sequence builds a correct tree, meaning that there will always be a CloseConstituent sometime after an OpenConstituent, for example Né)r(Únum_transitionsr5ÚCORRECTÚfnr;r:r=Údebugr?r8r&Útupler%r9)r@Úpred_transitionrrÚgold_transitionÚ repair_typeÚrepairrrr Ú fix_erroris&    &  ÿ zDynamicOracle.fix_errorN)Ú__name__Ú __module__Ú __qualname__rArLrrrr r/[s r/)Ú collectionsrÚnumpyrÚ,stanza.models.constituency.parse_transitionsrrrrr#r+r,r.r/rrrr Ús