
    9i                     B    S SK Jr  S SKJr  S SKJr  S\\   4S jrS rg)    )List)
OutputKeys)DialogStateTrackingPipeline	pipelinesc                    SS K n[        U5      n0 /n0 n[        U 5       Hd  u  pgUR                  U5        XU-     " UUS.5      n[	        UR                  U5      5        UR                  U[        R                     0 /5        Mf     g )Nr   )utterhistory_states)	jsonlen	enumerateupdateprintdumpsextendr   OUTPUT)		test_caser   r
   pipelines_lenr	   r   stepitemresults	            d/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/utils/nlp/space/utils_dst.py tracking_and_print_dialog_statesr      s    	NMTNE	*
T-/0	2
  	djj !vj&7&78"=> +    c                 .   / nU  H}  n[        U[        5      (       aE  UR                  UR                  5        VVs0 s H  u  pEXER	                  U5      _M     snn5        M]  UR                  UR	                  U5      5        M     [        U5      $ s  snnf )N)
isinstancedictappenditemstotuple)batchdevicebatch_on_deviceelementkvs         r   batch_to_devicer'      s    Ogt$$""$]]_.,TQ DDL,./ ""7::f#56  !!	.s   BN)typingr   modelscope.outputsr   modelscope.pipelines.nlpr   r   r'    r   r   <module>r,      s&     ) @?"#>??&	"r   