&`i |ddlZddlmZmZddlmZerddlmZddlm Z ddl m Z eGddZ dS) N) TYPE_CHECKINGOptional) PublicAPIAlgorithmConfig) RolloutWorker) SamplerInputc eZdZdZe ddeededdeded fd Zed ed fd Z e edZ e edZ dS) IOContextzClass containing attributes to pass to input/output class constructors. RLlib auto-sets these attributes when constructing input/output classes, such as InputReaders and OutputWriters. Nrlog_dirconfigr worker_indexworkerrcddlm}|ptj|_|p3||dud|_||_||_ dS)aInitializes a IOContext object. Args: log_dir: The logging directory to read from/write to. config: The (main) AlgorithmConfig object. worker_index: When there are multiple workers created, this uniquely identifies the current worker. 0 for the local worker, >0 for any of the remote workers. worker: The RolloutWorker object reference. rrN)actions_in_input_normalized)train_batch_size) %ray.rllib.algorithms.algorithm_configrosgetcwdr offline_datatrainingr rr)selfr r rrrs p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/offline/io_context.py__init__zIOContext.__init__s$ JIIIII-")++ ' 1 1 > >(.$!?! ! (A( & & ) returnr c|jjS)aiReturns the RolloutWorker's SamplerInput object, if any. Returns None if the RolloutWorker has no SamplerInput. Note that local workers in case there are also one or more remote workers by default do not create a SamplerInput object. Returns: The RolloutWorkers' SamplerInput object or None if none exists. )rsamplerrs rdefault_sampler_inputzIOContext.default_sampler_input3s{""rc8|jdiS)N input_configr getr s rr#zIOContext.input_config@s{~r222rc8|jdiS)N output_configr$r s rr'zIOContext.output_configEs{333r)NNrN) __name__ __module__ __qualname____doc__rrstrintrr!propertyr#r'rrr r s "&.2,0 #*+  ) Y< #x'? # # #Y #33YX344YX444rr ) rtypingrrray.rllib.utils.annotationsrrr#ray.rllib.evaluation.rollout_workerrray.rllib.evaluation.samplerr r r/rrr4s ********111111:EEEEEEAAAAAA999999 ;4;4;4;4;4;4;4 ;4;4;4r