Refactor: move task env imports to tasks/__init__.py

[yuecideng]

Description

Move explicit task environment imports from embodichain/lab/gym/envs/__init__.py into a newly created embodichain/lab/gym/envs/tasks/__init__.py.

The from .tasks import * wildcard import in envs/__init__.py now properly pulls in all task environments via the tasks package __init__.py, which:

• Keeps the envs/__init__.py clean with only 4 wildcard imports
• Centralizes all task environment registrations in the tasks/ package
• Preserves @register_env decorator side effects (gym registration still fires at import time)
• Defines an explicit __all__ in tasks/__init__.py for clear exports

Dependencies: None

Type of change

• Bug fix (non-breaking change which fixes an issue)
• Enhancement (non-breaking change which improves an existing functionality)
• New feature (non-breaking change which adds functionality)
• Breaking change (existing functionality will not work without user modification)
• Documentation update

Screenshots

N/A

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.

🤖 Generated with Claude Code

[Copilot]

Pull request overview

Refactors environment import/registration wiring by centralizing task-environment imports into the tasks package so envs/__init__.py can stay minimal while still triggering registration side effects on import.

Changes:

• Added embodichain/lab/gym/envs/tasks/__init__.py to import and re-export task environments with an explicit __all__.
• Removed explicit task-environment imports from embodichain/lab/gym/envs/__init__.py, relying on from .tasks import *.

Reviewed changes

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

File	Description
embodichain/lab/gym/envs/tasks/__init__.py	Centralizes task env imports/exports and defines __all__ for wildcard exports.
embodichain/lab/gym/envs/__init__.py	Simplifies top-level env imports to wildcard-import the tasks package.

---

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

[Copilot]

Adding tasks/__init__.py turns embodichain.lab.gym.envs.tasks from a (previously implicit) namespace package into a regular package, so any import of a submodule like embodichain.lab.gym.envs.tasks.rl will first execute this file and eagerly import/register all task envs. This can significantly increase import time and introduces broad import-time side effects for code paths that only need RL utilities (e.g., agents/rl/train.py). Consider moving the eager env imports into a dedicated module that is imported only from envs/__init__.py (or a register_all_tasks() function called there), keeping tasks/__init__.py lightweight to avoid forcing all task imports on every tasks.* import.

[Copilot]

This refactor changes import/registration wiring but there’s no test ensuring task env registrations still occur via import embodichain.lab.gym.envs / from embodichain.lab.gym.envs import * (and that exports in __all__ resolve). Since the repo already has pytest coverage for lab/gym/envs, please add a lightweight test that imports embodichain.lab.gym.envs and asserts gymnasium.spec(...) (or the internal registry) contains a representative env id like SimpleTask-v1 / PushCubeRL.