ray.rllib.policy.sample_batch.MultiAgentBatch.decompress_if_needed

MultiAgentBatch.decompress_if_needed(columns: Set[str] = frozenset({'new_obs', 'obs'})) → MultiAgentBatch

Decompresses each policy batch (per column), if already compressed.

Parameters:

columns – Set of column names to decompress.

Returns:

Self.