&`i>ddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZddlZddlmZddlmZddlmZdd lmZm Z dd l!m"Z"dd l#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*dd l+m,Z,m-Z-e rddl.Z/ddlm0Z0ddl1m2Z2ej3e4Z5dZ6e-GddZ7dee7ee7effdZ8dS)N)partial)Path) PicklingError) TYPE_CHECKINGAnyCallableDictListMappingOptionalSequenceTypeUnion)RpcError)StorageContext)DEFAULT_STORAGE_PATH)CheckpointConfig SyncConfig) TuneError)is_function_trainableregister_trainable)CombinedStopperFunctionStopperStopperTimeoutStopper) Deprecated DeveloperAPI)PlacementGroupFactory)Trialc|sdx}}nt|tr|rd}d}nt|tr|x}}nt|trHt |dkr/t dt ||\}}n/t dt|||fS)zsValidate ``tune.RunConfig``'s ``log_to_file`` parameter. Return validated relative stdout and stderr filenames.NstdoutstderrzIf you pass a Sequence to `log_to_file` it has to have a length of 2 (for stdout and stderr, respectively). The Sequence you passed has length {}.zqYou can pass a boolean, a string, or a Sequence of length 2 to `log_to_file`, but you passed something else ({}).) isinstanceboolstrr len ValueErrorformattype) log_to_file stdout_file stderr_files r/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/experiment/experiment.py_validate_log_to_filer/*s  $(( kk K & & ;   K % % $// kk K * * {  q 55;VC stop num_samples time_budget_sNFr)r3r5configresources_per_trialr4 storage_pathstorage_filesystem sync_configcheckpoint_configtrial_name_creatortrial_dirname_creatorr+export_formats max_failuresrestore local_dirnamerunr3r5r7r8rr4r9r:zpyarrow.fs.FileSystemr;r<r=rr>r+r?r@rArBct| tr tdi| } n| p t} t|r,| jrt d| jrt d t||_ nQ#t$rD}|j tj jkr#tdt!j|d}~wwxYw|st%j|}|pt(}||| | ||_t.d|j|pi}t|ts t dt3|dd|_i}|snt|t6rz'Experiment.__init__..s'JJJ!:a3I3IJAJJJr0c,g|]}t|SrH)r*rIs r.rLz'Experiment.__init__..s 7 7 7Qa 7 7 7r0zIf you pass a list as the `stop` argument to `tune.RunConfig()`, each element must be an instance of `tune.stopper.Stopper`. Got .zaProvided stop object must be either a dict, a function, or a subclass of `ray.tune.Stopper`. Got zInvalid stop criteria: z". Must be a callable or dict. Got )rDr3r5r7r8r4r<r=r>r+r?r@rAstoragerH)(r$dictrrcheckpoint_at_endr(checkpoint_frequencyr2register_if_needed_run_identifierrrpc_coderay_raylet#GRPC_STATUS_CODE_RESOURCE_EXHAUSTEDr traceback format_excrget_experiment_dir_namer_storage_context_clsrOloggerdebugr*_stopperlistrcallableris_valid_functionrrr/r expanduserabsoluteas_posixspec)selfrCrDr3r5r7r8r4r9r:r;r<r=r>r+r?r@rArBestopping_criteria bad_stoppers stopper_typesr,r-rfs r.__init__zExperiment.__init__fs'2 ' . . H 0 E E3D E E   1 G5E5G5G   % %  2  M !5  , #-#@#@#E#ED    zS[LLLD ,5+?+A+A DD  ?!9#>>D#;'; 00%1# $ 1    Et|EEFFF2&$'' ST&\\SSS      d # # JJtJJJL  7 7$ 7 7 7  D3@DDD ,T2DMM d # #  $   d^^ 066  / 5 5 D'**  $  =/3Dzz=== 7$77)-d777   >} > /M>-#@#@!! !/} = = #8#E#E ['%*#6&!2"4%:'5,2(DKUW ((**3355>>@@@QU|!  $ s/B C?CCrfc:d|vrtdd|vr-|di|d<|d|dd<|d=d|vr0t|dtrt d i|d|d<d|vr0t|dtrt d i|d|d<t j|}|d} |||fi|}n9#t$r,}tdtj |d|d }~wwxYw|S) zGenerates an Experiment object from JSON. Args: name: Name of Experiment. spec: JSON configuration of experiment. rDzNo trainable specified!envr7r;r<z>Failed to load the following Tune experiment specification: zd. Please check that the arguments are valid. Experiment creation failed with the following error: NrH) rgetr$rPrrcopydeepcopypop TypeErrorpppformat)clsrCrf run_valueexprhs r. from_jsonzExperiment.from_jsonsv   566 6 D==!XXh33DN$(KDN5 !U D Z]0CT%J%J ","C"CtM/B"C"CD  $ & &:d;N6OQU+V+V &(8(U(U4@S;T(U(UD$ %}T""HHUOO  #dI....CC    &(j&6&6        s C"" D,'DD run_objectcXddlm}t|tst||r|St|tst |rd}t |dr|j}nt |dr+|j}|dkrd}n| drd}nw|}ntt|tr2t |d r"t |j dr |j j}n-t d ||Std ) aGet Trainable name. Args: run_object: Trainable to run. If string, assumes it is an ID and does not modify it. Otherwise, returns a string corresponding to the run_object name. Returns: A string representing the trainable identifier. Raises: TuneError: if ``run_object`` passed in is invalid. rDomainDEFAULT_name__name__zlambda<funcz(No name detected on trainable. Using {}.z*Improper 'run' - not string nor trainable.)ray.tune.search.sampler}r$r&r*rahasattrrr startswithrrr]warningr)r)rvrzr}rCfn_names r.get_trainable_namezExperiment.get_trainable_namesX 211111 j# & & J*Z*H*H J   D ) ) JXj-A-A JDz7++ X!'Z00 X$-j((#DD'',,#$DD"DD:w// XJ// XJOZ88 X "/IPPQUVVWWWKHII Ir0ctddlm}t|tr|St||rtd|S||} t||nG#ttf$r3}d}t|t|dz|zdd}~wwxYw|S)aRegisters Trainable or Function at runtime. Assumes already registered if run_object is a string. Also, does not inspect interface of given run_object. Args: run_object: Trainable to run. If string, assumes it is an ID and does not modify it. Otherwise, returns a string corresponding to the run_object name. Returns: A string representing the trainable identifier. rr|z0Not registering trainable. Resolving as variant.zOther options: -Try reproducing the issue by calling `pickle.dumps(trainable)`. -If the error is typing-related, try removing the type annotations and try again. N) rr}r$r&r]rrrrsrr*)rvrzr}rCrh extra_msgs r.rSzExperiment.register_if_neededGs 211111 j# & &    F + +  NNM N N N %%j11 > tZ 0 0 0 0=) > > >6 $q''#a&&3,233 = > s A11B5.B00B5c|jSN)r_rgs r.stopperzExperiment.stopperks }r0returnc|jjSr)rOexperiment_driver_staging_pathrs r. local_pathzExperiment.local_pathos |::r0zReplaced by `local_path`c td)Nz(Use `local_path` instead of `local_dir`.DeprecationWarningrs r.rBzExperiment.local_dirss!!KLLLr0c|jjSr)rOexperiment_fs_pathrs r. remote_pathzExperiment.remote_pathys |..r0c|jp|jSr)rrrs r.pathzExperiment.path}s24?2r0c6|jdS)Nr<)rfrors r.r<zExperiment.checkpoint_configsy}}0111r0c td)Nz-Use `local_path` instead of `checkpoint_dir`.rrs r.checkpoint_dirzExperiment.checkpoint_dirs!!PQQQr0c|jS)z7Returns a string representing the trainable identifier.)rTrs r.run_identifierzExperiment.run_identifiers ##r0cNfdjDS)zReturns the spec dict with only the public-facing keys. Intended to be used for passing information to callbacks, Searchers and Schedulers. c.i|]\}}|jv||SrH) PUBLIC_KEYS)rJkvrgs r. z*Experiment.public_spec..s,LLLAa4;K6K6K16K6K6Kr0)rfitemsrs`r. public_speczExperiment.public_specs,MLLL!2!2LLLLr0))r __module__ __qualname____doc__rrr\r&rrrr r rr%intfloatdatetime timedeltar rrrPrr rl classmethodryrrSpropertyrrrrBrrr<rrrrHr0r.r2r2Fs$6;::K)SWIM+/ &*@D9=EIAEDH!-1!%#'/MMMM3$& 'M uWgxg8L/MMNO M  c5(2D&D EF Mc3h(M# '#uUC%899:!M" ('C(@A#M$%M&!*'M()M*#+M.C=/MMMM^$S$$$$[$L(JE#x2E,F(J(J(J[(JT!E#x2E,F!!![!FX;HSM;;;X;Z*++MM,+XM/Xc]///X/3hsm333X322X2Z*++RR,+XR$$X$MT#s(^MMMXMMMr0r2 experimentsc|}|g}nMt|tr|g}n4t|turd|D}t|t urGt d|Dr.t|dkrt dn"td ||S)a%Produces a list of Experiment objects. Converts input from dict, single experiment, or list of experiments to list of experiments. If input is None, will return an empty list. Arguments: experiments: Experiments to run. Returns: List of experiments. NcJg|] \}}t||!SrH)r2ry)rJrCrfs r.rLz/_convert_to_experiment_list..s9   1;tJ t , ,   r0c3@K|]}t|tVdSr)r$r2)rJrxs r. z._convert_to_experiment_list..s,%V%Vcjj&A&A%V%V%V%V%V%Vr0r6zeRunning with multiple concurrent experiments. All experiments will be using the same SearchAlgorithm.zInvalid argument: {}) r$r2r*rPrr`allr'r]inforr))rexp_lists r._convert_to_experiment_listrsH K , , = k  d " "  ?J?P?P?R?R     H~~#%V%VX%V%V%V"V"V x==1   KKJ    .55kBBCCC Or0)9rprloggingpprintrtrY functoolsrpathlibrpicklertypingrrrr r r r r rrrVray.exceptionsrray.train._internal.storagerray.train.constantsrray.tunerrray.tune.errorrray.tune.registryrrray.tune.stopperrrrrray.util.annotationsrr pyarrow.fspyarrowrray.tune.experimentr getLoggerrr]r/r2rrHr0r.rs|                           ######66666644444411111111$$$$$$GGGGGGGGVVVVVVVVVVVV99999999*......))))))  8 $ $$$$8PMPMPMPMPMPMPMPMf #U:tJ?OQU3U-V######r0