<filename>reward/env/__init__.py
"""
The environment is the world that the agent interacts with, it could be a game,
a physics engine or anything you would like. It should receive and execute an
action and return to the agent the next observation and a reward.
"""

from .base_env import BaseEnv
from .gym_env import GymEnv
from .atari_env import AtariEnv
from .roboschool_env import RoboschoolEnv
import reward.env.wrappers

__all__ = ["BaseEnv", "GymEnv", "RoboschoolEnv", "AtariEnv"]

<filename>reward/env/__init__.py
"""
The environment is the world that the agent interacts with, it could be a game,
a physics engine or anything you would like. It should receive and execute an
action and return to the agent the next observation and a reward.
"""

from .base_env import BaseEnv
from .gym_env import GymEnv
from .atari_env import AtariEnv
from .roboschool_env import RoboschoolEnv
import reward.env.wrappers

__all__ = ["BaseEnv", "GymEnv", "RoboschoolEnv", "AtariEnv"]

The environment is the world that the agent interacts with, it could be a game, a physics engine or anything you would like. It should receive and execute an action and return to the agent the next observation and a reward.