admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +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
src/lerobot/policies/wall_x/__init__.py
+1 -3
View File
@@ -15,7 +15,5 @@
# limitations under the License. # limitations under the License.
from .configuration_wall_x import WallXConfig from .configuration_wall_x import WallXConfig
from .modeling_wall_x import WallXPolicy
from .processor_wall_x import make_wall_x_pre_post_processors
__all__ = ["WallXConfig", "WallXPolicy", "make_wall_x_pre_post_processors"] __all__ = ["WallXConfig"]
tests/policies/wall_x/test_wallx.py
+3 -5
View File
@@ -37,11 +37,9 @@ if WALLX_AVAILABLE:
import torch import torch
from lerobot.policies.factory import make_policy_config from lerobot.policies.factory import make_policy_config
from lerobot.policies.wall_x import ( from lerobot.policies.wall_x import WallXConfig
WallXConfig, from lerobot.policies.wall_x.modeling_wall_x import WallXPolicy
WallXPolicy, from lerobot.policies.wall_x.processor_wall_x import make_wall_x_pre_post_processors
make_wall_x_pre_post_processors,
)
from lerobot.utils.random_utils import set_seed from lerobot.utils.random_utils import set_seed