`evorl.sample_batch`¶

Module Contents¶

Classes¶

`Episode`	The container for an episode trajectory.
`SampleBatch`	Data container for trajectory data.

API¶

class evorl.sample_batch.Episode[source]¶

Bases: evorl.types.PyTreeData

The container for an episode trajectory.

trajectory: evorl.sample_batch.SampleBatch¶: None

property valid_mask: chex.Array¶

class evorl.sample_batch.SampleBatch[source]¶

Bases: evorl.types.PyTreeData, evorl.types.PyTreeArrayMixin

Data container for trajectory data.

actions: chex.ArrayTree | None¶: None

dones: chex.Array | None¶: None

extras: evorl.types.ExtraInfo | None¶: None

next_obs: chex.Array | None¶: None

obs: chex.ArrayTree | None¶: None

rewards: evorl.types.Reward | evorl.types.RewardDict | None¶: None