From eadd4c0856ca4cbf62f8a522f0ad2d6907172b0c Mon Sep 17 00:00:00 2001 From: Pepijn Date: Wed, 17 Dec 2025 16:29:18 +0100 Subject: [PATCH] only export WallXConfig from wall_x package to avoid peft import in CI --- src/lerobot/policies/wall_x/__init__.py | 4 +--- tests/policies/wall_x/test_wallx.py | 8 +++----- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/src/lerobot/policies/wall_x/__init__.py b/src/lerobot/policies/wall_x/__init__.py index 16fd2c8ab..be4542fde 100644 --- a/src/lerobot/policies/wall_x/__init__.py +++ b/src/lerobot/policies/wall_x/__init__.py @@ -15,7 +15,5 @@ # limitations under the License. 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"] diff --git a/tests/policies/wall_x/test_wallx.py b/tests/policies/wall_x/test_wallx.py index 2951ca8b8..5226d590a 100644 --- a/tests/policies/wall_x/test_wallx.py +++ b/tests/policies/wall_x/test_wallx.py @@ -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