testing with cfg
Browse files
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).
|