Commit Graph

5 Commits

Author SHA1 Message Date
Maximellerbach 972b32fc99 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-31 22:34:48 +02:00
Maximellerbach a4f59cd5be adding more tests to ensure good coverage 2026-05-31 22:34:48 +02:00
Maximellerbach b370da1813 adding guards to avoid needing transformers and diffusers for type checking and basic tests 2026-05-31 22:34:47 +02:00
Maximellerbach a72b3bd2f4 linting 2026-05-31 22:31:42 +02:00
ginwind 3536d17cfb (feat)policies: add VLA-JEPA 2026-05-31 22:31:42 +02:00