Pi{ ddlmZddlZddlZddlmZmZddlZddl m Z ddl m Z ddlmZdedeeje jfdeeje jfd ed eejejff d ZdS) N)CallableUnion)ExceptionWrapper) QUEUE_TIMEOUT worker_idin_qout_qudf stop_eventc8tjd |r|rdS |dt \}}n#t j$rY]wxYwt|tr| ||fdnut|tr| ||fdnF ||}n #t$rtd}YnwxYw| ||fd) aX_apply_udf assumes in_q emits tuples of (x, idx) where x is the payload, idx is the index of the result, potentially used for maintaining ordered outputs. For every input it pulls, a tuple (y, idx) is put on the out_q where the output of udf(x), an ExceptionWrapper, or StopIteration (if it pulled StopIteration from in_q). rT)blocktimeoutF)rz in _apply_udf)whereN) torchset_num_threadsis_setemptygetrqueueEmpty isinstancerput StopIteration Exception)rr r r r itemidxys n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchdata/nodes/_apply_udf.py _apply_udfr sI !-      4::<<  E t]CCID##{    H  d, - - - IItSkI / / / / m , , - IItSkI / / / / <CII < < <$?;;; < IIq#heI , , ,'-s$A!!A32A3 C!!C>=C>)multiprocessing.synchronize synchronizepython_mp_synchronizer threadingtypingrrrtorch.multiprocessingmultiprocessingmp torch._utilsr constantsrintQueueEventr rr0s<;;;;; """""""" """"""))))))$$$$$$!-!-  RX% &!- bh& '!-  !- io'<'BBC !-!-!-!-!-!-r/