Commit f4c64b60 authored by peterd-NV's avatar peterd-NV Committed by Kelly Guo

Adds GR1 env import to required scripts (#321)

# Description

<!--
Thank you for your interest in sending a pull request. Please make sure
to check the contribution guidelines.

Link:
https://isaac-sim.github.io/IsaacLab/main/source/refs/contributing.html
-->

Adds missing GR1 env import to scripts that may use the humanoid
environment.

Fixes case where running replay/train/play scripts gives
"Isaac-PickPlace-GR1T2-Abs-v0" not found error

## Type of change

<!-- As you go through the list, delete the ones that are not
applicable. -->

- Bug fix (non-breaking change which fixes an issue)


<!--
Example:

| Before | After |
| ------ | ----- |
| _gif/png before_ | _gif/png after_ |

To upload images to a PR -- simply drag and drop an image while in edit
mode and it should upload the image directly. You can then paste that
source into the above before/after sections.
-->

## Checklist

- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [ ] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there

<!--
As you go through the checklist above, you can mark something as done by
putting an x character in it

For example,
- [x] I have done this task
- [ ] I have not done this task
-->
parent 86c1d372
...@@ -66,6 +66,9 @@ import torch ...@@ -66,6 +66,9 @@ import torch
import robomimic.utils.file_utils as FileUtils import robomimic.utils.file_utils as FileUtils
import robomimic.utils.torch_utils as TorchUtils import robomimic.utils.torch_utils as TorchUtils
if args_cli.enable_pinocchio:
import isaaclab_tasks.manager_based.manipulation.pick_place # noqa: F401
from isaaclab_tasks.utils import parse_env_cfg from isaaclab_tasks.utils import parse_env_cfg
......
...@@ -84,6 +84,7 @@ from robomimic.utils.log_utils import DataLogger, PrintLogger ...@@ -84,6 +84,7 @@ from robomimic.utils.log_utils import DataLogger, PrintLogger
# Isaac Lab imports (needed so that environment is registered) # Isaac Lab imports (needed so that environment is registered)
import isaaclab_tasks # noqa: F401 import isaaclab_tasks # noqa: F401
import isaaclab_tasks.manager_based.manipulation.pick_place # noqa: F401
def normalize_hdf5_actions(config: Config, log_dir: str) -> str: def normalize_hdf5_actions(config: Config, log_dir: str) -> str:
......
...@@ -65,6 +65,9 @@ import torch ...@@ -65,6 +65,9 @@ import torch
from isaaclab.devices import Se3Keyboard from isaaclab.devices import Se3Keyboard
from isaaclab.utils.datasets import EpisodeData, HDF5DatasetFileHandler from isaaclab.utils.datasets import EpisodeData, HDF5DatasetFileHandler
if args_cli.enable_pinocchio:
import isaaclab_tasks.manager_based.manipulation.pick_place # noqa: F401
import isaaclab_tasks # noqa: F401 import isaaclab_tasks # noqa: F401
from isaaclab_tasks.utils.parse_cfg import parse_env_cfg from isaaclab_tasks.utils.parse_cfg import parse_env_cfg
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment