mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 18:26:11 +00:00
feat(envs): add env plugin discovery
- Add 'lerobot_env_' to third-party plugin discovery prefixes, completing the plugin system for all component types (robots, cameras, teleoperators, policies, and now environments). External packages named lerobot_env_* can self-register EnvConfig subclasses on import, enabling --env.type= resolution without lerobot code changes.
This commit is contained in:
@@ -216,9 +216,15 @@ def register_third_party_plugins() -> None:
|
|||||||
|
|
||||||
This function uses `importlib.metadata` to find packages installed in the environment
|
This function uses `importlib.metadata` to find packages installed in the environment
|
||||||
(including editable installs) starting with 'lerobot_robot_', 'lerobot_camera_',
|
(including editable installs) starting with 'lerobot_robot_', 'lerobot_camera_',
|
||||||
'lerobot_teleoperator_', or 'lerobot_policy_' and imports them.
|
'lerobot_teleoperator_', 'lerobot_policy_', or 'lerobot_env_' and imports them.
|
||||||
"""
|
"""
|
||||||
prefixes = ("lerobot_robot_", "lerobot_camera_", "lerobot_teleoperator_", "lerobot_policy_")
|
prefixes = (
|
||||||
|
"lerobot_robot_",
|
||||||
|
"lerobot_camera_",
|
||||||
|
"lerobot_teleoperator_",
|
||||||
|
"lerobot_policy_",
|
||||||
|
"lerobot_env_",
|
||||||
|
)
|
||||||
imported: list[str] = []
|
imported: list[str] = []
|
||||||
failed: list[str] = []
|
failed: list[str] = []
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user