memorax.utils.Timestep

class memorax.utils.Timestep

Bases: object

Timestep(obs: jax.jaxlib._jax.Array, action: jax.jaxlib._jax.Array, reward: jax.jaxlib._jax.Array, done: jax.jaxlib._jax.Array)

obs: Array

action: Array

reward: Array

done: Array

to_sequence()

from_sequence()

__init__(obs, action, reward, done)

Parameters:

• obs (Array)

• action (Array)

• reward (Array)

• done (Array)

Return type:

None

replace(**updates)

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