admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-16 00:59:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

only export WallXConfig from wall_x package to avoid peft import in CI

Browse Source
This commit is contained in:
Pepijn
2025-12-17 16:29:18 +01:00
committed by Michel Aractingi
parent 3434a5d5df
commit eadd4c0856
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/wall_x/test_wallx.py
+3 -5
View File
@@ -37,11 +37,9 @@ if WALLX_AVAILABLE:
import torch
from lerobot.policies.factory import make_policy_config
from lerobot.policies.wall_x import (
WallXConfig,
WallXPolicy,
make_wall_x_pre_post_processors,
)
from lerobot.policies.wall_x import WallXConfig
from lerobot.policies.wall_x.modeling_wall_x import WallXPolicy
from lerobot.policies.wall_x.processor_wall_x import make_wall_x_pre_post_processors
from lerobot.utils.random_utils import set_seed