renamed to generic isaaclab wrapper
Browse files
env.py
CHANGED
|
@@ -29,7 +29,7 @@ import numpy as np
|
|
| 29 |
import torch
|
| 30 |
import yaml
|
| 31 |
|
| 32 |
-
from lerobot.envs.isaaclab_arena import
|
| 33 |
from lerobot.utils.errors import IsaacLabArenaCameraKeyError, IsaacLabArenaStateKeyError
|
| 34 |
|
| 35 |
# Hub constants for downloading additional files
|
|
@@ -246,7 +246,7 @@ def _create_isaaclab_env(config: dict, n_envs: int) -> dict[str, dict[int, gym.v
|
|
| 246 |
task = f"Complete the {environment.replace('_', ' ')} task."
|
| 247 |
|
| 248 |
# Wrap and return
|
| 249 |
-
wrapped_env =
|
| 250 |
raw_env,
|
| 251 |
episode_length=config.get("episode_length", 300),
|
| 252 |
task=task,
|
|
|
|
| 29 |
import torch
|
| 30 |
import yaml
|
| 31 |
|
| 32 |
+
from lerobot.envs.isaaclab_arena import IsaacLabEnvWrapper, cleanup_isaaclab
|
| 33 |
from lerobot.utils.errors import IsaacLabArenaCameraKeyError, IsaacLabArenaStateKeyError
|
| 34 |
|
| 35 |
# Hub constants for downloading additional files
|
|
|
|
| 246 |
task = f"Complete the {environment.replace('_', ' ')} task."
|
| 247 |
|
| 248 |
# Wrap and return
|
| 249 |
+
wrapped_env = IsaacLabEnvWrapper(
|
| 250 |
raw_env,
|
| 251 |
episode_length=config.get("episode_length", 300),
|
| 252 |
task=task,
|