admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-29 12:39:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(rtc): share Pi05 training helpers

Browse Source
This commit is contained in:
Pepijn
2026-07-17 17:41:20 +02:00
parent d9c05b76aa
commit 12740f6be0
5 changed files with 7 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/common/test_vla_utils.py
+2 -2
View File
@@ -56,8 +56,8 @@ def test_create_sinusoidal_pos_embedding_matches_openpi_formula():
def test_create_sinusoidal_pos_embedding_validation():
with pytest.raises(ValueError, match="divisible by 2"):
create_sinusoidal_pos_embedding(torch.zeros(2), 7, 4e-3, 4.0, device=torch.device("cpu"))
with pytest.raises(ValueError, match="batch_size"):
create_sinusoidal_pos_embedding(torch.zeros(2, 2), 8, 4e-3, 4.0, device=torch.device("cpu"))
with pytest.raises(ValueError, match="must have shape"):
create_sinusoidal_pos_embedding(torch.zeros(2, 2, 2), 8, 4e-3, 4.0, device=torch.device("cpu"))
def test_make_att_2d_masks_docstring_cases():