ray/rllib/offline/output_writer.py

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


@PublicAPI
class OutputWriter:
    """Writer object for saving experiences from policy evaluation."""

    @PublicAPI
    def write(self, sample_batch):
        """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):
        pass