"""Imports wrappers

TODO: DEPRECATE. This pattern imports unecessary modules, creating unnecessary
software dependencies. Just import the specific module instead.
"""

from rlkit.envs.wrappers.discretize_env import DiscretizeEnv
from rlkit.envs.wrappers.history_env import HistoryEnv
# from rlkit.envs.wrappers.image_mujoco_env import ImageMujocoEnv
# from rlkit.envs.wrappers.image_mujoco_env_with_obs import ImageMujocoWithObsEnv
from rlkit.envs.wrappers.normalized_box_env import NormalizedBoxEnv
from rlkit.envs.proxy_env import ProxyEnv
from rlkit.envs.wrappers.reward_wrapper_env import RewardWrapperEnv
from rlkit.envs.wrappers.stack_observation_env import StackObservationEnv


__all__ = [
    'DiscretizeEnv',
    'HistoryEnv',
    # 'ImageMujocoEnv',
    # 'ImageMujocoWithObsEnv',
    'NormalizedBoxEnv',
    'ProxyEnv',
    'RewardWrapperEnv',
    'StackObservationEnv',
]
