mirror of
https://github.com/wassname/ray.git
synced 2026-07-07 17:03:03 +08:00
Eric Liang
29 lines
706 B
Python
29 lines
706 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
from ray.rllib.utils.annotations import override
|
|
from ray.rllib.utils.annotations import PublicAPI
|
|
|
|
|
|
@PublicAPI
|
|
class OutputWriter(object):
|
|
"""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
|