nvkartik commited on
Commit
d6b9500
·
1 Parent(s): 0efe22f

renamed to generic isaaclab wrapper

Browse files
Files changed (1) hide show
  1. env.py +2 -2
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 IsaacLabArenaEnvWrapper, cleanup_isaaclab
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 = IsaacLabArenaEnvWrapper(
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,