<reponame>zhihanyang2022/CleanRL<filename>offpcc/domains/dmc_acrobot.py
import dmc2gym
from domains.wrappers import ConcatObs
from domains.wrappers_for_image import Normalize255Image, GrayscaleImage, ConcatImages


def mdp():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=[], track_prev_action=False, frame_skip=5)


def p():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=['velocity'], track_prev_action=False, frame_skip=5)


def va():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=['orientations'], track_prev_action=True, frame_skip=5)


def p_concat5():
    return ConcatObs(p(), 5)


def va_concat10():
    return ConcatObs(va(), 10)

import dmc2gym
from domains.wrappers import ConcatObs
from domains.wrappers_for_image import Normalize255Image, GrayscaleImage, ConcatImages


def mdp():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=[], track_prev_action=False, frame_skip=5)


def p():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=['velocity'], track_prev_action=False, frame_skip=5)


def va():
    return dmc2gym.make(domain_name="acrobot", task_name="swingup", keys_to_exclude=['orientations'], track_prev_action=True, frame_skip=5)


def p_concat5():
    return ConcatObs(p(), 5)


def va_concat10():
    return ConcatObs(va(), 10)