ray/rllib/env/atari_wrappers.py

25 lines
863 B
Python

from ray.rllib.env.wrappers.atari_wrappers import is_atari, \
get_wrapper_by_cls, MonitorEnv, NoopResetEnv, ClipRewardEnv, \
FireResetEnv, EpisodicLifeEnv, MaxAndSkipEnv, WarpFrame, FrameStack, \
FrameStackTrajectoryView, ScaledFloatFrame, wrap_deepmind
from ray.rllib.utils.deprecation import deprecation_warning
deprecation_warning(
old="ray.rllib.env.atari_wrappers....",
new="ray.rllib.env.wrappers.atari_wrappers....",
error=False,
)
is_atari = is_atari
get_wrapper_by_cls = get_wrapper_by_cls
MonitorEnv = MonitorEnv
NoopResetEnv = NoopResetEnv
ClipRewardEnv = ClipRewardEnv
FireResetEnv = FireResetEnv
EpisodicLifeEnv = EpisodicLifeEnv
MaxAndSkipEnv = MaxAndSkipEnv
WarpFrame = WarpFrame
FrameStack = FrameStack
FrameStackTrajectoryView = FrameStackTrajectoryView
ScaledFloatFrame = ScaledFloatFrame
wrap_deepmind = wrap_deepmind