Update env.py
Browse files
env.py
CHANGED
|
@@ -46,10 +46,15 @@ def _download_and_import(filename: str):
|
|
| 46 |
|
| 47 |
|
| 48 |
try:
|
| 49 |
-
from .
|
|
|
|
| 50 |
except ImportError:
|
|
|
|
| 51 |
_isaaclab_wrapper = _download_and_import("isaaclab_env_wrapper.py")
|
| 52 |
IsaacLabEnvWrapper = _isaaclab_wrapper.IsaacLabEnvWrapper
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
|
| 55 |
def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
try:
|
| 49 |
+
from .errors import IsaacLabArenaCameraKeyError, IsaacLabArenaStateKeyError
|
| 50 |
+
from .isaaclab_env_wrapper import IsaacLabEnvWrapper, cleanup_isaaclab
|
| 51 |
except ImportError:
|
| 52 |
+
_errors = _download_and_import("errors.py")
|
| 53 |
_isaaclab_wrapper = _download_and_import("isaaclab_env_wrapper.py")
|
| 54 |
IsaacLabEnvWrapper = _isaaclab_wrapper.IsaacLabEnvWrapper
|
| 55 |
+
cleanup_isaaclab = _isaaclab_wrapper.cleanup_isaaclab
|
| 56 |
+
IsaacLabArenaCameraKeyError = _errors.IsaacLabArenaCameraKeyError
|
| 57 |
+
IsaacLabArenaStateKeyError = _errors.IsaacLabArenaStateKeyError
|
| 58 |
|
| 59 |
|
| 60 |
def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):
|