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

moving and renaming files to have a cleaner file tree

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-06-30 11:01:33 +00:00
parent 343ab5f99c
commit 218f8a1735
13 changed files with 54 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/fastwam/test_fastwam_policy.py
+2 -2
View File
@@ -257,7 +257,7 @@ def test_from_pretrained_uses_base_loader_and_skips_wan_backbone(monkeypatch, tm
raise AssertionError("from_pretrained must not initialize or download the Wan2.2 backbone")
monkeypatch.setattr(
"lerobot.policies.fastwam.modular_fastwam.FastWAM.from_wan22_pretrained",
"lerobot.policies.fastwam.wan.modular.FastWAM.from_wan22_pretrained",
fail_if_wan_pretrained_is_loaded,
)
@@ -348,7 +348,7 @@ def test_vae_adapter_empty_build_encode_decode_shapes():
pytest.importorskip("diffusers")
from diffusers import AutoencoderKLWan
from lerobot.policies.fastwam.wan_adapters import WanVideoVAE38
from lerobot.policies.fastwam.wan.adapters import WanVideoVAE38
# Production always loads a real pretrained VAE from the diffusers repo; here we
# build the same architecture with random weights and dummy standardization stats