ray.rllib.policy.sample_batch.SampleBatch.decompress_if_needed

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

Decompresses data buffers (per column if not compressed) in place.

Parameters:

columns – The columns to decompress. Default: Only decompress the obs and new_obs columns.

Returns:

This very (now uncompressed) SampleBatch.