ray.rllib.offline.io_context.IOContext#

class ray.rllib.offline.io_context.IOContext(log_dir: str | None = None, config: AlgorithmConfig | None = None, worker_index: int = 0, worker: RolloutWorker | None = None)[source]#

Class containing attributes to pass to input/output class constructors.

RLlib auto-sets these attributes when constructing input/output classes, such as InputReaders and OutputWriters.

Methods

__init__

Initializes a IOContext object.

default_sampler_input

Returns the RolloutWorker's SamplerInput object, if any.

Attributes

input_config

output_config