:py:mod:`rofunc.learning.RofuncRL.utils.elegantrl_utils`
========================================================

.. py:module:: rofunc.learning.RofuncRL.utils.elegantrl_utils

.. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils
   :allowtitles:

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

Classes
~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`ElegantRLIsaacGymEnvWrapper <rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper>`
     - .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper
          :summary:

Functions
~~~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`setup <rofunc.learning.RofuncRL.utils.elegantrl_utils.setup>`
     - .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.setup
          :summary:

API
~~~

.. py:class:: ElegantRLIsaacGymEnvWrapper(env, cfg, should_print=False)
   :canonical: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper

   .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper

   .. rubric:: Initialization

   .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper.__init__

   .. py:method:: reset() -> torch.Tensor
      :canonical: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper.reset

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper.reset

   .. py:method:: step(actions: torch.Tensor) -> typing.Tuple[torch.Tensor, torch.Tensor, torch.Tensor, typing.Dict]
      :canonical: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper.step

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.ElegantRLIsaacGymEnvWrapper.step

.. py:function:: setup(custom_args, eval_mode=False)
   :canonical: rofunc.learning.RofuncRL.utils.elegantrl_utils.setup

   .. autodoc2-docstring:: rofunc.learning.RofuncRL.utils.elegantrl_utils.setup