rename IsaacLabArenaEnvWrapper to IsaacLabEnvWrapper.
Browse files
env.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import logging
|
| 2 |
from typing import Any
|
| 3 |
-
from lerobot.envs.
|
| 4 |
|
| 5 |
|
| 6 |
def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):
|
|
@@ -29,7 +29,7 @@ def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):
|
|
| 29 |
config_path=config_path
|
| 30 |
)
|
| 31 |
|
| 32 |
-
wrapped_env =
|
| 33 |
raw_env,
|
| 34 |
episode_length=episode_length,
|
| 35 |
task=task,
|
|
|
|
| 1 |
import logging
|
| 2 |
from typing import Any
|
| 3 |
+
from lerobot.envs.isaaclab import IsaacLabEnvWrapper
|
| 4 |
|
| 5 |
|
| 6 |
def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):
|
|
|
|
| 29 |
config_path=config_path
|
| 30 |
)
|
| 31 |
|
| 32 |
+
wrapped_env = IsaacLabEnvWrapper(
|
| 33 |
raw_env,
|
| 34 |
episode_length=episode_length,
|
| 35 |
task=task,
|