OmniSafe Wrapper
================

.. currentmodule:: omnisafe.envs.wrapper

.. autosummary::

    TimeLimit
    AutoReset
    ObsNormalize
    RewardNormalize
    CostNormalize
    ActionScale
    Unsqueeze

Time Limit Wrapper
------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: TimeLimit
        :members:
        :private-members:

Auto Reset Wrapper
------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: AutoReset
        :members:
        :private-members:


Observation Normalization Wrapper
---------------------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: ObsNormalize
        :members:
        :private-members:

Reward Normalization Wrapper
----------------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: RewardNormalize
        :members:
        :private-members:

Cost Normalization Wrapper
--------------------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: CostNormalize
        :members:
        :private-members:

Action Scale
------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: ActionScale
        :members:
        :private-members:

Action Repeat
-------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: ActionRepeat
        :members:
        :private-members:

Unsqueeze Wrapper
-----------------

.. card::
    :class-header: sd-bg-success sd-text-white
    :class-card: sd-outline-success  sd-rounded-1

    Documentation
    ^^^

    .. autoclass:: Unsqueeze
        :members:
        :private-members:
