mani_skill.utils.wrappers.visual_encoders
=========================================

.. py:module:: mani_skill.utils.wrappers.visual_encoders


Classes
-------

.. autoapisummary::

   mani_skill.utils.wrappers.visual_encoders.VisualEncoderWrapper


Module Contents
---------------

.. py:class:: VisualEncoderWrapper(env, encoder, encoder_config=dict())

   Bases: :py:obj:`gymnasium.ObservationWrapper`


   Modify observations from :meth:`Env.reset` and :meth:`Env.step` using :meth:`observation` function.

   If you would like to apply a function to only the observation before
   passing it to the learning code, you can simply inherit from :class:`ObservationWrapper` and overwrite the method
   :meth:`observation` to implement that transformation. The transformation defined in that method must be
   reflected by the :attr:`env` observation space. Otherwise, you need to specify the new observation space of the
   wrapper by setting :attr:`self.observation_space` in the :meth:`__init__` method of your wrapper.


   .. py:method:: observation(obs)

      Returns a modified observation.

      :param observation: The :attr:`env` observation

      :returns: The modified observation



   .. py:attribute:: base_env
      :type:  mani_skill.envs.sapien_env.BaseEnv