Commit Graph

5 Commits

Author SHA1 Message Date
Maximellerbach 1f80d91607 update VLA-JEPA tests for arch changes and action_is_pad
- Switch conftest to use `action_model_type="DiT-test"` now that
  `action_num_heads` / `action_attention_head_dim` have been removed.
- Add action_head tests covering fully-padded loss (zero) and equivalence
  of action_is_pad=None vs all-zeros mask.
- Remove obsolete `test_native_to_lerobot_wm_only` test.
2026-05-15 14:34:41 +02:00
Maximellerbach ddaff399b5 adding more tests to ensure good coverage 2026-05-13 15:55:04 +02:00
Maximellerbach 96a8efddf0 adding guards to avoid needing transformers and diffusers for type checking and basic tests 2026-05-12 11:01:41 +02:00
Maximellerbach 2e9ba42e1b linting 2026-05-12 11:01:41 +02:00
ginwind 3144029814 (feat)policies: add VLA-JEPA 2026-05-12 11:01:41 +02:00