evorl.ec.optimizers.evox_wrapper

Module Contents

Classes

EvoXAlgoState

EvoXAlgorithmAdapter

Adapter class to convert EvoX algorithms to EvoRL optimizers.

API

class evorl.ec.optimizers.evox_wrapper.EvoXAlgoState[source]

Bases: evorl.types.PyTreeData

algo_state: evox.State

None

init_step: bool

‘pytree_field(…)’

class evorl.ec.optimizers.evox_wrapper.EvoXAlgorithmAdapter[source]

Bases: evorl.ec.optimizers.ec_optimizer.EvoOptimizer

Adapter class to convert EvoX algorithms to EvoRL optimizers.

algorithm: evox.Algorithm

None

ask(state: evorl.ec.optimizers.evox_wrapper.EvoXAlgoState) → tuple[chex.ArrayTree, evorl.ec.optimizers.evox_wrapper.EvoXAlgoState][source]
init(key: chex.PRNGKey) → evorl.ec.optimizers.evox_wrapper.EvoXAlgoState[source]
param_vec_spec: evorl.utils.ec_utils.ParamVectorSpec

None

tell(state: evorl.ec.optimizers.evox_wrapper.EvoXAlgoState, fitnesses: chex.Array) → tuple[evorl.types.PyTreeDict, evorl.ec.optimizers.evox_wrapper.EvoXAlgoState][source]