biM!PUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZddlZddlmZddlmZmZmZdd lmZmZdd lmZdd lmZdd l m!Z!d Z"ej e#e$jJe"eejLejNZ(ejRZ*dZ+Gdde,Z-iZ.de/d<ddZ0 ddl1Z2e2jfjidZ5dZ7dZ8dZ9 d d!dZ:dZ;ddZt|j|jffSN)getattr __objclass____name__)ms O/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/multiprocessing.py_reduce_method_descriptorrs Q^^QZZ0 00)protocolc>tjjSr)multiprocessingcurrent_processidentrr_process_get_idr!$s  * * , 2 22rc(eZdZdZdZdZdZdZy)RemoteExceptionzXRemote Exception Contains the exception and traceback from a remotely run task c ||_||_yr) exception traceback)selfr%r&s r__init__zRemoteException.__init__;s""rcLt|jdz|jzS)Nz Traceback --------- )strr%r&r's r__str__zRemoteException.__str__?s 4>>"%AADNNRRrc tttt|t |j zt|j zSr)sortedsetdirtypelist__dict__r%r+s r__dir__zRemoteException.__dir__Bs6c#d4j/D,??#dnnBUUVWWrcz tj||S#t$rt|j|cYSwxYwr)object__getattribute__AttributeErrorrr%)r'keys r __getattr__zRemoteException.__getattr__Es: 0**45 5 04>>3/ / 0s ::N)r __module__ __qualname____doc__r(r,r4r:r rrr#r#5s #SX0rr#z&dict[type[Exception], type[Exception]] exceptionsc*t|tvrtt|}|||S t|jjtt|fdt|i}|tt|<|||S#t $r|cYSwxYw)z6Metaclass that wraps exception type in RemoteExceptionexception_type)r1r> __class__rr# TypeError)exctbtyps rremote_exceptionrFOs CyJc#3|  && $s),!49-C %(JtCy !sB<  J sAB BBc|Srr rDs r_pack_tracebackrIfs rcJdjtj|S)N)joinr& format_tbrHs rrIrIkswwy**2.//rct||}|r)rF)rCrDs rr r nssB' rctj\}}}t|} |||f}|S#t$r8}tj\}}}t|}|||f}Yd}~|Sd}~wwxYwr)sysexc_inforI Exception)edumpsexc_type exc_value exc_tracebackrDresults rpack_exceptionrYssw),&Hi  'B 2w M  -0\\^*)] ] +2w M  s 1 A2-A--A2zThe 'multiprocessing.context' configuration option will be ignored on Python 2 and on Windows, because they each only support a single context. ctjdd}tjdk(r|dk7rt t t tStj|S)z+Return the current multiprocessing context.zmultiprocessing.contextspawnwin32) r getrPplatformr_CONTEXT_UNSUPPORTED UserWarningr get_context) context_names rrarasK::7AL ||w 7 " %{ 3**<88rc |xstjdd}|xstjdd}|xstjdd}|xstjddxst}|`tjjddvrd tjd<t } t t| }t|| | }d } n>| td t|tjjr t|}d} t|dr|j!}t#|}t%||\} } |rt'| || \}} n| }|xstjddxst(}|xstjddxst*} t-|j.|j0||ft2||t4t6|d| }| r|j9|S#| r|j9wwxYw)aMultiprocessed get function appropriate for Bags Parameters ---------- dsk : dict dask graph keys : object or list Desired results from graph num_workers : int Number of worker processes (defaults to number of cores) func_dumps : function Function to use for function serialization (defaults to cloudpickle.dumps) func_loads : function Function to use for function deserialization (defaults to cloudpickle.loads) optimize_graph : bool If True [default], `fuse` is applied to the graph before computation. pool : Executor or Pool Some sort of `Executor` or `Pool` to use initializer: function Ignored if ``pool`` has been set. Function to initialize a worker process before running any tasks in it. chunksize: int, optional Size of chunks to use when dispatching work. Defaults to 6 as some batching is helpful. If -1, will be computed to evenly divide ready work across workers. chunksizepoolNzmultiprocessing.initializer num_workersPYTHONHASHSEED)N06640user_initializer) mp_context initializerTzThe ``initializer`` argument is ignored when ``pool`` is provided. The user should configure ``pool`` with the needed ``initializer`` on creation.F__dask_graph__ func_loads func_dumps)get_idrTloadsrYraise_exceptionrd)r r]rosenvironrarinitialize_worker_processrr isinstancerrfPoolr hasattrrorr r_loads_dumpsr submit _max_workersr!rYr shutdown)dskkeysrgrprqoptimize_graphrfrnrdkwargscontextcleanupdsk2 dependenciesdsk3rsrTrXs rr]r]sL7VZZ Q7I  +6::fd+DP,I4!PKMM4!@MIK | ::>>* +{ :,2BJJ' (-7+V " G   "   dO0055 6.t4Ds$%  " c Cc4D,!$l;l  B&**\48 BFE  B&**\48 BFE  KK       #)#      MMO M  MMO s !6G++Hc|tjjd}||jj yy)Nnumpy)rPmodulesr]randomseed)nps rdefault_initializerrs.  !B ~ rc,t||yy)zE Initialize a worker process before running any tasks in it. N)rrks rrwrws#$r)rCrRreturnrRr)NNNTNNN)rrrzSequence[Key] | Key)= __future__rcopyregrmultiprocessing.poolrupicklerPr&collections.abcrrconcurrent.futuresr functoolsrwarningsr cloudpickledaskr dask.localr r r dask.optimizationr r dask.systemr dask.typingr dask.utilsrrr1r/unionrTHIGHEST_PROTOCOLr|rsr{r!rRr#r>__annotations__rFtblib.pickling_supporttblibpickling_supportinstallrI ImportErrorrYr_rar]rrwr rrrs." -2FF(!"1 tCII 9: ""V-D-D E   3"0i0.68 27$! ""$  9" i i iX}0 s!D D%$D%