xpag.buffers.buffer.EpisodicBuffer

class EpisodicBuffer(buffer_size, sampler)

Bases: Buffer

Base class for episodic buffers

Methods

insert	Inserts a transition in the buffer
sample	Uses the sampler to returns a batch of transitions
store_done	Stores the episodes that are done

abstract insert(step)

Inserts a transition in the buffer

abstract sample(batch_size)

Uses the sampler to returns a batch of transitions

Return type:

Dict[str, Union[ndarray, Array]]

abstract store_done(done)

Stores the episodes that are done