Source code for ray.rllib.offline.output_writer

from ray.rllib.utils.annotations import override
from ray.rllib.utils.annotations import PublicAPI
from ray.rllib.utils.types import SampleBatchType


[docs]@PublicAPI class OutputWriter: """Writer object for saving experiences from policy evaluation."""
[docs] @PublicAPI def write(self, sample_batch: SampleBatchType): """Save a batch of experiences. Arguments: sample_batch: SampleBatch or MultiAgentBatch to save. """ raise NotImplementedError
class NoopOutput(OutputWriter): """Output writer that discards its outputs.""" @override(OutputWriter) def write(self, sample_batch: SampleBatchType): pass