mani_skill.vector.wrappers.sb3
==============================

.. py:module:: mani_skill.vector.wrappers.sb3


Classes
-------

.. autoapisummary::

   mani_skill.vector.wrappers.sb3.ManiSkillSB3VectorEnv


Functions
---------

.. autoapisummary::

   mani_skill.vector.wrappers.sb3.select_index_from_dict


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

.. py:class:: ManiSkillSB3VectorEnv(env)

   Bases: :py:obj:`stable_baselines3.common.vec_env.base_vec_env.VecEnv`


   A wrapper for to make ManiSkill parallel simulation compatible with SB3 VecEnv and auto adds the monitor wrapper


   .. py:method:: close()


   .. py:method:: env_is_wrapped(wrapper_class, indices = None)


   .. py:method:: env_method(method_name, *method_args, indices = None, **method_kwargs)


   .. py:method:: get_attr(attr_name, indices = None)


   .. py:method:: reset()


   .. py:method:: seed(seed = None)


   .. py:method:: set_attr(attr_name, value, indices = None)


   .. py:method:: step_async(actions)


   .. py:method:: step_wait()


   .. py:attribute:: _env


   .. py:attribute:: _last_seed
      :value: None



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



   .. py:attribute:: episode_lengths
      :type:  torch.Tensor


   .. py:attribute:: episode_returns
      :type:  torch.Tensor


   .. py:attribute:: episode_times
      :type:  torch.Tensor


   .. py:attribute:: t_start


   .. py:attribute:: total_steps
      :value: 0



.. py:function:: select_index_from_dict(data, i)