:py:mod:`rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp`
=====================================================================

.. py:module:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp

.. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp
   :allowtitles:

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

Classes
~~~~~~~

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

   * - :py:obj:`HumanoidAMP <rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP>`
     -

Functions
~~~~~~~~~

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

   * - :py:obj:`build_amp_observations <rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.build_amp_observations>`
     - .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.build_amp_observations
          :summary:

API
~~~

.. py:class:: HumanoidAMP(cfg, rl_device, sim_device, graphics_device_id, headless, virtual_screen_capture, force_render)
   :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP

   Bases: :py:obj:`rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid.Humanoid`

   .. py:class:: StateInit
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit

      Bases: :py:obj:`enum.Enum`

      .. py:attribute:: Default
         :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Default
         :value: 0

         .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Default

      .. py:attribute:: Start
         :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Start
         :value: 1

         .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Start

      .. py:attribute:: Random
         :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Random
         :value: 2

         .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Random

      .. py:attribute:: Hybrid
         :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Hybrid
         :value: 3

         .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.StateInit.Hybrid

   .. py:method:: post_physics_step()
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.post_physics_step

   .. py:method:: get_num_amp_obs()
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.get_num_amp_obs

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.get_num_amp_obs

   .. py:property:: amp_observation_space
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.amp_observation_space

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.amp_observation_space

   .. py:method:: fetch_amp_obs_demo(num_samples)
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.fetch_amp_obs_demo

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.fetch_amp_obs_demo

   .. py:method:: build_amp_obs_demo(motion_ids, motion_times0)
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.build_amp_obs_demo

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.build_amp_obs_demo

   .. py:method:: reset_idx(env_ids)
      :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.HumanoidAMP.reset_idx

.. py:function:: build_amp_observations(root_pos: Tensor, root_rot: Tensor, root_vel: Tensor, root_ang_vel: Tensor, dof_pos: Tensor, dof_vel: Tensor, key_body_pos: Tensor, local_root_obs: bool, root_height_obs: bool, dof_obs_size: int, dof_offsets: List[int]) -> Tensor
   :canonical: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.build_amp_observations

   .. autodoc2-docstring:: rofunc.learning.RofuncRL.tasks.isaacgymenv.ase.humanoid_amp.build_amp_observations