Skip to content

Adds blueprint environment for Franka stacking mimic #1944

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 19 commits into from
Mar 10, 2025

Conversation

chengronglai
Copy link
Contributor

@chengronglai chengronglai commented Feb 26, 2025

Description

This pull request refactors the core functionality of generate_dataset.py into separate helper modules. The goal is to make the associated Jupyter notebook leaner.

  • Move functions in generate_dataset to isaaclab_mimic/datagen/generation.py
  • Add utils.py for utility function for the notebook
  • Add dependency for ipywidgets

Type of change

  • New feature (non-breaking change which adds functionality)

Checklist

  • [ x] I have run the pre-commit checks with ./isaaclab.sh --format
  • I have made corresponding changes to the documentation
  • 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

* Add a jupyter notebook for generate_dataset.py
* Move functions in generate_dataset to isaaclab_mimic/datagen/generation.py
* Add utils.py for utility function for the notebook
* Add dependency for ipywidgets
* Add randomizable_params in EventTermCfg for franka stack_joint_pos_env_cfg
* Clearer description of randomizable parameters for users
* Use FloatRangeSlider and FloatSlider to adjust value
* Update scene with reset and stepping for better visual feedback
@kellyguo11
Copy link
Contributor

looks like the docs are failing to build, might need to add ipywidgets to conf.py. will ipywidgets also need to be added to setup.py?

chengronglai and others added 2 commits March 5, 2025 09:01
* Add proper type hints to all functions in datagen/utils.py and datagen/generation.py
# 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
-->

Please include a summary of the change and which issue is fixed. Please
also include relevant motivation and context.
List any dependencies that are required for this change.

Fixes # (issue)

<!-- As a practice, it is recommended to open an issue to have
discussions on the proposed pull request.
This makes it easier for the community to keep track of what is being
developed or added, and if a given feature
is demanded by more than one party. -->

## 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)
- New feature (non-breaking change which adds functionality)
- Breaking change (fix or feature that would cause existing
functionality to not work as expected)
- This change requires a documentation update

## Screenshots

Please attach before and after screenshots of the change if applicable.

<!--
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

- [ ] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [ ] 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
-->

---------

Signed-off-by: Kelly Guo <[email protected]>
Co-authored-by: Peter Du <[email protected]>
Co-authored-by: Kelly Guo <[email protected]>
@kellyguo11 kellyguo11 requested a review from pascal-roth as a code owner March 5, 2025 19:12
@kellyguo11 kellyguo11 changed the title Chengrongl/randomization selection data generation Adds blueprint environment for Franka stacking mimic Mar 6, 2025
@kellyguo11 kellyguo11 merged commit 27cec08 into main Mar 10, 2025
3 of 4 checks passed
@kellyguo11 kellyguo11 deleted the chengrongl/randomization_selection_data_generation branch March 10, 2025 21:31
jtigue-bdai pushed a commit that referenced this pull request Apr 14, 2025
# Description

This pull request refactors the core functionality of
generate_dataset.py into separate helper modules. The goal is to make
the associated Jupyter notebook leaner.

* Move functions in generate_dataset to
isaaclab_mimic/datagen/generation.py
* Add utils.py for utility function for the notebook
* Add dependency for ipywidgets

## Type of change

- New feature (non-breaking change which adds functionality)

## 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
- [ ] 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
-->

---------

Signed-off-by: Kelly Guo <[email protected]>
Co-authored-by: Jean-Francois-Lafleche <[email protected]>
Co-authored-by: Peter Du <[email protected]>
Co-authored-by: Kelly Guo <[email protected]>
Co-authored-by: Kelly Guo <[email protected]>
ToxicNS pushed a commit to ToxicNS/IsaacLab that referenced this pull request Apr 24, 2025
# Description

This pull request refactors the core functionality of
generate_dataset.py into separate helper modules. The goal is to make
the associated Jupyter notebook leaner.

* Move functions in generate_dataset to
isaaclab_mimic/datagen/generation.py
* Add utils.py for utility function for the notebook
* Add dependency for ipywidgets

## Type of change

- New feature (non-breaking change which adds functionality)

## 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
- [ ] 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
-->

---------

Signed-off-by: Kelly Guo <[email protected]>
Co-authored-by: Jean-Francois-Lafleche <[email protected]>
Co-authored-by: Peter Du <[email protected]>
Co-authored-by: Kelly Guo <[email protected]>
Co-authored-by: Kelly Guo <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants