JumanjiEnv

torchrl.envs.libs.jumanji.JumanjiEnv(*args, **kwargs)

Jumanji environment wrapper.

Examples

>>> env = JumanjiEnv(env_name="Snake-6x6-v0", frame_skip=4)
>>> td = env.rand_step()
>>> print(td)
>>> print(env.available_envs)