mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-06 09:37:06 +00:00
fix(scripts): register openarm_mini for train CLI parsing
Import openarm_mini in lerobot_train to trigger teleoperator config registration, preventing lerobot scripts from failing to parse configs after the OpenArm Mini teleoperator addition. Made-with: Cursor
This commit is contained in:
@@ -38,6 +38,7 @@ from lerobot.policies.factory import make_policy, make_pre_post_processors
|
||||
from lerobot.policies.pretrained import PreTrainedPolicy
|
||||
from lerobot.rl.wandb_utils import WandBLogger
|
||||
from lerobot.scripts.lerobot_eval import eval_policy_all
|
||||
from lerobot.teleoperators import openarm_mini # noqa: F401
|
||||
from lerobot.utils.import_utils import register_third_party_plugins
|
||||
from lerobot.utils.logging_utils import AverageMeter, MetricsTracker
|
||||
from lerobot.utils.random_utils import set_seed
|
||||
|
||||
Reference in New Issue
Block a user