Organize table in wrap_env docstrings

Toni-SM · Aug 27, 2024 · 48459c4 · 48459c4
1 parent 967e85e
commit 48459c4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/skrl/envs/wrappers/jax/__init__.py b/skrl/envs/wrappers/jax/__init__.py
@@ -44,6 +44,10 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - ``"gymnasium"``
                         * - Brax
                           - ``"brax"``
+                        * - Isaac Lab
+                          - ``"isaaclab"``
+                        * - Isaac Lab (multi-agent as single-agent)
+                          - ``"isaaclab-single-agent"``
                         * - Isaac Gym preview 2
                           - ``"isaacgym-preview2"``
                         * - Isaac Gym preview 3
@@ -52,10 +56,6 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - ``"isaacgym-preview4"``
                         * - Omniverse Isaac Gym
                           - ``"omniverse-isaacgym"``
-                        * - Isaac Lab
-                          - ``"isaaclab"``
-                        * - Isaac Lab (multi-agent as single-agent)
-                          - ``"isaaclab-single-agent"``
 
                     .. list-table:: Multi-agent environments |br|
                         :header-rows: 1
@@ -64,10 +64,10 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - Wrapper tag
                         * - Petting Zoo
                           - ``"pettingzoo"``
-                        * - Bi-DexHands
-                          - ``"bidexhands"``
                         * - Isaac Lab
                           - ``"isaaclab-multi-agent"``
+                        * - Bi-DexHands
+                          - ``"bidexhands"``
     :type wrapper: str, optional
     :param verbose: Whether to print the wrapper type (default: ``True``)
     :type verbose: bool, optional

diff --git a/skrl/envs/wrappers/torch/__init__.py b/skrl/envs/wrappers/torch/__init__.py
@@ -50,10 +50,12 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - ``"gymnasium"``
                         * - DeepMind
                           - ``"dm"``
-                        * - Robosuite
-                          - ``"robosuite"``
                         * - Brax
                           - ``"brax"``
+                        * - Isaac Lab
+                          - ``"isaaclab"``
+                        * - Isaac Lab (multi-agent as single-agent)
+                          - ``"isaaclab-single-agent"``
                         * - Isaac Gym preview 2
                           - ``"isaacgym-preview2"``
                         * - Isaac Gym preview 3
@@ -62,10 +64,8 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - ``"isaacgym-preview4"``
                         * - Omniverse Isaac Gym
                           - ``"omniverse-isaacgym"``
-                        * - Isaac Lab
-                          - ``"isaaclab"``
-                        * - Isaac Lab (multi-agent as single-agent)
-                          - ``"isaaclab-single-agent"``
+                        * - Robosuite
+                          - ``"robosuite"``
 
                     .. list-table:: Multi-agent environments |br|
                         :header-rows: 1
@@ -74,10 +74,10 @@ def wrap_env(env: Any, wrapper: str = "auto", verbose: bool = True) -> Union[Wra
                           - Wrapper tag
                         * - Petting Zoo
                           - ``"pettingzoo"``
-                        * - Bi-DexHands
-                          - ``"bidexhands"``
                         * - Isaac Lab
                           - ``"isaaclab-multi-agent"``
+                        * - Bi-DexHands
+                          - ``"bidexhands"``
     :type wrapper: str, optional
     :param verbose: Whether to print the wrapper type (default: ``True``)
     :type verbose: bool, optional