Support customized get_data_class target modules

[yuecideng]

Description

Same as PR title

Type of change

• Enhancement (non-breaking change which improves an existing functionality)

Checklist

• I have run the black . command to format the code base.
• I have made corresponding changes to the documentation
• I have added tests that prove my fix is effective or that my feature works
• Dependencies have been updated, if applicable.

[Copilot]

Pull request overview

This PR adds support for customized target modules in the get_data_class function, allowing callers to search for dataset classes in additional modules beyond the default ones. However, the PR also includes unrelated changes to the preview_sensor_data method in the embodied environment.

Changes:

• Added extra_modules parameter to get_data_class function to support custom dataset module locations
• Refactored module list to use DEFAULT_DATA_MODULES constant instead of inline list
• Added save parameter to preview_sensor_data method to optionally save images instead of displaying them
• Changed default visualization method from "plt" to "cv2" in preview_sensor_data

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 3 comments.

File	Description
embodichain/data/dataset.py	Adds extra_modules parameter to get_data_class and refactors to use DEFAULT_DATA_MODULES constant
embodichain/lab/gym/envs/embodied_env.py	Adds save parameter and changes default method to "cv2" in preview_sensor_data

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.