rlightning.buffer.sampler

Sampling strategies for selecting data from buffers.

class rlightning.buffer.sampler.AllDataSampler

Bases: BaseSampler

Sample all data from the buffer without replacement.

sample(batch_size: int, data_size: int, shuffle: bool | None = False)

Return all indices, optionally shuffled.

class rlightning.buffer.sampler.BaseSampler(replacement: bool)

Bases: ABC

Abstract base class for buffer sampling strategies.

abstractmethod sample(batch_size: int, data_size: int, shuffle: bool | None = False)

Sample indices from a dataset.

class rlightning.buffer.sampler.BatchSampler

Bases: BaseSampler

Sample a batch of data from the buffer without replacement.

sample(batch_size: int, data_size: int, shuffle: bool | None = False)

Sample a batch of indices without replacement.

class rlightning.buffer.sampler.UniformSampler

Bases: BaseSampler

Uniformly sample data from the buffer with replacement.

sample(batch_size: int, data_size: int, shuffle: bool | None = False)

Sample indices uniformly.