evorl.envs.jumanji

Module Contents

Classes

JumanjiAdapter

Adapter for Jumanji environments.

Functions

API

class evorl.envs.jumanji.JumanjiAdapter(env: jumanji.env.Environment)

Bases: evorl.envs.env.EnvAdapter

Adapter for Jumanji environments.

property action_space: evorl.envs.space.Space
property obs_space: evorl.envs.space.Space
reset(key: chex.PRNGKey) evorl.envs.env.EnvState
step(state: evorl.envs.env.EnvState, action: evorl.types.Action) evorl.envs.env.EnvState
evorl.envs.jumanji.create_jumanji_env(env_name: str, **kwargs) evorl.envs.jumanji.JumanjiAdapter
evorl.envs.jumanji.jumanji_specs_to_evorl_space(spec)