Wrappers
========

Environment wrappers compatible with the ``RLEnv`` protocol.

Episode Wrapper
---------------

.. automodule:: nnx_ppo.wrappers.episode_wrapper
   :members:
   :special-members: __init__
   :show-inheritance:

Reward Scaling Wrapper
----------------------

.. automodule:: nnx_ppo.wrappers.reward_scaling_wrapper
   :members:
   :special-members: __init__
   :show-inheritance: