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

pre-commit cleanup

Browse Source
This commit is contained in:
Maximellerbach
2026-05-21 13:16:57 +02:00
parent 553c217ee2
commit 1066d037d5
6 changed files with 38 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/vla_jepa/test_world_model.py
+3 -1
View File
@@ -43,7 +43,9 @@ def _make_predictor(
],
)
def test_predictor_output_shape(batch: int, num_steps: int, tokens_per_frame: int, embed_dim: int) -> None:
predictor = _make_predictor(embed_dim=embed_dim, action_embed_dim=_ACTION_EMBED_DIM, tokens_per_frame=tokens_per_frame)
predictor = _make_predictor(
embed_dim=embed_dim, action_embed_dim=_ACTION_EMBED_DIM, tokens_per_frame=tokens_per_frame
)
frame_tokens = torch.randn(batch, num_steps * tokens_per_frame, embed_dim)
action_tokens = torch.randn(batch, num_steps * 2, _ACTION_EMBED_DIM)
out = predictor(frame_tokens, action_tokens)