torchrl/envs/libs/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

from .brax import BraxEnv, BraxWrapper
from .dm_control import DMControlEnv, DMControlWrapper
from .envpool import MultiThreadedEnv, MultiThreadedEnvWrapper
from .gym import (
    gym_backend,
    GymEnv,
    GymWrapper,
    MOGymEnv,
    MOGymWrapper,
    register_gym_spec_conversion,
    set_gym_backend,
)
from .habitat import HabitatEnv
from .isaacgym import IsaacGymEnv, IsaacGymWrapper
from .jumanji import JumanjiEnv, JumanjiWrapper
from .meltingpot import MeltingpotEnv, MeltingpotWrapper
from .openml import OpenMLEnv
from .openspiel import OpenSpielEnv, OpenSpielWrapper
from .pettingzoo import PettingZooEnv, PettingZooWrapper
from .robohive import RoboHiveEnv
from .smacv2 import SMACv2Env, SMACv2Wrapper
from .unity_mlagents import UnityMLAgentsEnv, UnityMLAgentsWrapper
from .vmas import VmasEnv, VmasWrapper