memorax.utils.Timestep

class memorax.utils.Timestep

Bases: object

Timestep(obs: jax.Array | None = None, action: jax.Array | None = None, reward: jax.Array | None = None, done: jax.Array | None = None)

obs: Array | None = None

action: Array | None = None

reward: Array | None = None

done: Array | None = None

to_sequence()

Return type:

Self

from_sequence()

Return type:

Self

__init__(obs=None, action=None, reward=None, done=None)

Parameters:

• obs (Array | None)

• action (Array | None)

• reward (Array | None)

• done (Array | None)

Return type:

None

replace(**updates)

Returns a new object replacing the specified fields with new values.