nvkartik commited on
Commit
3f9b161
·
1 Parent(s): af4bcff

testing with cfg

Browse files
Files changed (1) hide show
  1. env.py +2 -0
env.py CHANGED
@@ -29,6 +29,7 @@ import numpy as np
29
  import torch
30
  import yaml
31
 
 
32
  from lerobot.envs.isaaclab import IsaacLabEnvWrapper, cleanup_isaaclab
33
  from lerobot.utils.errors import IsaacLabArenaCameraKeyError, IsaacLabArenaStateKeyError
34
 
@@ -260,6 +261,7 @@ def _create_isaaclab_env(config: dict, n_envs: int) -> dict[str, dict[int, gym.v
260
  def make_env(
261
  n_envs: int = 1,
262
  use_async_envs: bool = False,
 
263
  **kwargs: Any,
264
  ) -> dict[str, dict[int, gym.vector.VectorEnv]]:
265
  """Create IsaacLab Arena environments (EnvHub-compatible API).
 
29
  import torch
30
  import yaml
31
 
32
+ from lerobot.envs.configs import EnvConfig
33
  from lerobot.envs.isaaclab import IsaacLabEnvWrapper, cleanup_isaaclab
34
  from lerobot.utils.errors import IsaacLabArenaCameraKeyError, IsaacLabArenaStateKeyError
35
 
 
261
  def make_env(
262
  n_envs: int = 1,
263
  use_async_envs: bool = False,
264
+ cfg: EnvConfig | None = None,
265
  **kwargs: Any,
266
  ) -> dict[str, dict[int, gym.vector.VectorEnv]]:
267
  """Create IsaacLab Arena environments (EnvHub-compatible API).