fix: wall-oss bugs

[YushunXiang]

Title

Updated WallXPolicy.__init__ to accept **kwargs and pass them to the superclass initializer, enabling future extensibility.

Type / Scope

• Type: Bug

Summary / Motivation

This pull request introduces a small update to the initialization method of the WallXPolicy class to allow passing additional keyword arguments. This change improves flexibility for subclassing and integration.

How was this tested

• Manual checks / dataset runs performed.

How to run locally (reviewer)

• Run the relevant tests:

  pytest -vs tests/policies/wall_x/test_wallx.py 

• Run a quick example or CLI (if applicable):

  python src/lerobot/scripts/lerobot_train.py \
      --dataset.repo_id=your_dataset \
      --policy.type=wall_x \
      --output_dir=./outputs/wallx_training \
      --job_name=wallx_training \
      --policy.repo_id=your_repo_id \
      --policy.pretrained_name_or_path=x-square-robot/wall-oss-flow \
      --policy.prediction_mode=diffusion \
      --policy.attn_implementation=eager \
      --steps=3000 \
      --policy.device=cuda \
      --batch_size=32

Checklist (required before merge)

• Linting/formatting run (pre-commit run -a)
• All tests pass locally (pytest)
• Documentation updated
• CI is green

Reviewer notes

• Anything the reviewer should focus on (performance, edge-cases, specific files) or general notes.
• Anyone in the community is free to review the PR.

@wut19 @pkooij

[Copilot]

Pull request overview

This pull request adds **kwargs parameter to the WallXPolicy.__init__ method to enable passing additional keyword arguments through the initialization chain. While this aims to improve flexibility for future extensibility, the implementation is inconsistent with existing patterns in the codebase.

Key Changes

• Modified WallXPolicy.__init__ signature to accept **kwargs
• Updated the superclass __init__ call to forward **kwargs

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.