ray.rllib.policy.sample_batch.MultiAgentBatch.compress#

MultiAgentBatch.compress(bulk: bool = False, columns: Set[str] = frozenset({'new_obs', 'obs'})) → None[source]#

Compresses each policy batch (per column) in place.

Parameters:

  • bulk – Whether to compress across the batch dimension (0) as well. If False will compress n separate list items, where n is the batch size.

  • columns – Set of column names to compress.