Functional Env

Converting Jax-based Functional environments to standard Env

class gymnasium.envs.functional_jax_env.FunctionalJaxEnv(func_env: FuncEnv, metadata: dict[str, Any] | None = None, render_mode: str | None = None, spec: EnvSpec | None = None)

A conversion layer for jax-based environments.

Initialize the environment from a FuncEnv.

reset(*, seed: int | None = None, options: dict | None = None)

Resets the environment using the seed.

step(action: ActType)

Steps through the environment using the action.

render()

Returns the render state if render_mode is “rgb_array”.