from gym.envs.registration import load
from .wrappers import WarpFrame, ScaledFloatFrame, MaxAndSkipEnv, FrameStack
from .normalized_env import NormalizedActionWrapper

def mujoco_wrapper(entry_point, **kwargs):
    # Load the environment from its entry point
    env_cls = load(entry_point)
    env = env_cls(**kwargs)
    # Normalization wrapper
    env = NormalizedActionWrapper(env)
    return env

def universe_wrapper(entry_point, **kwargs):
	# Load the environment from its entry point
    env_cls = load(entry_point)
    env = env_cls(**kwargs)

    # Preprocessing wrappers
    env = MaxAndSkipEnv(env, skip=4)
    env = WarpFrame(env, dim=84, rgb=True)
    env = ScaledFloatFrame(env)
    return env
from gym.envs.registration import load
from .wrappers import WarpFrame, ScaledFloatFrame, MaxAndSkipEnv, FrameStack
from .normalized_env import NormalizedActionWrapper

def mujoco_wrapper(entry_point, **kwargs):
    # Load the environment from its entry point
    env_cls = load(entry_point)
    env = env_cls(**kwargs)
    # Normalization wrapper
    env = NormalizedActionWrapper(env)
    return env

def universe_wrapper(entry_point, **kwargs):
	# Load the environment from its entry point
    env_cls = load(entry_point)
    env = env_cls(**kwargs)

    # Preprocessing wrappers
    env = MaxAndSkipEnv(env, skip=4)
    env = WarpFrame(env, dim=84, rgb=True)
    env = ScaledFloatFrame(env)
    return env
# Load the environment from its entry point # Normalization wrapper # Load the environment from its entry point # Preprocessing wrappers