memorax.algorithms.StreamACState

class memorax.algorithms.StreamACState

Bases: object

StreamACState(step: int, update_step: int, timestep: memorax.utils.timestep.Timestep, env_state: gymnax.environments.environment.EnvState, actor_params: flax.core.frozen_dict.FrozenDict[str, typing.Any], actor_traces: flax.core.frozen_dict.FrozenDict[str, typing.Any], actor_v: flax.core.frozen_dict.FrozenDict[str, typing.Any], actor_carry: jax.Array, critic_params: flax.core.frozen_dict.FrozenDict[str, typing.Any], critic_traces: flax.core.frozen_dict.FrozenDict[str, typing.Any], critic_v: flax.core.frozen_dict.FrozenDict[str, typing.Any], critic_carry: jax.Array)

step: int

update_step: int

timestep: Timestep

env_state: EnvState

actor_params: FrozenDict[str, Any]

actor_traces: FrozenDict[str, Any]

actor_v: FrozenDict[str, Any]

actor_carry: Array

critic_params: FrozenDict[str, Any]

critic_traces: FrozenDict[str, Any]

critic_v: FrozenDict[str, Any]

critic_carry: Array

__init__(step, update_step, timestep, env_state, actor_params, actor_traces, actor_v, actor_carry, critic_params, critic_traces, critic_v, critic_carry)

Parameters:

• step (int)

• update_step (int)

• timestep (Timestep)

• env_state (EnvState)

• actor_params (FrozenDict[str, Any])

• actor_traces (FrozenDict[str, Any])

• actor_v (FrozenDict[str, Any])

• actor_carry (Array)

• critic_params (FrozenDict[str, Any])

• critic_traces (FrozenDict[str, Any])

• critic_v (FrozenDict[str, Any])

• critic_carry (Array)

Return type:

None

replace(**updates)

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