admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 02:59:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

removing missleading future_action_window_size to just use chunk_size

Browse Source
This commit is contained in:
Maximellerbach
2026-05-18 18:14:13 +02:00
parent 4e85e23350
commit ed6978fd18
6 changed files with 4 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/vla_jepa/conftest.py
-1
View File
@@ -62,7 +62,6 @@ def make_config(
device="cpu",
chunk_size=action_horizon,
n_action_steps=min(N_ACTION_STEPS, action_horizon),
future_action_window_size=action_horizon - 1,
action_dim=action_dim,
state_dim=state_dim,
num_video_frames=num_video_frames,
tests/policies/vla_jepa/test_configuration.py
+1 -7
View File
@@ -18,12 +18,7 @@ def test_delta_indices() -> None:
def test_n_action_steps_exceeds_chunk_size_raises() -> None:
with pytest.raises(ValueError, match="n_action_steps"):
VLAJEPAConfig(chunk_size=4, n_action_steps=8, future_action_window_size=3)
def test_future_window_exceeds_chunk_size_raises() -> None:
with pytest.raises(ValueError, match="predicted action horizon"):
VLAJEPAConfig(chunk_size=4, n_action_steps=4, future_action_window_size=4)
VLAJEPAConfig(chunk_size=4, n_action_steps=8)
def test_too_few_video_frames_raises() -> None:
@@ -31,7 +26,6 @@ def test_too_few_video_frames_raises() -> None:
VLAJEPAConfig(
chunk_size=16,
n_action_steps=16,
future_action_window_size=15,
num_video_frames=2,
jepa_tubelet_size=2, # needs >= 4 frames (2 for current, 2 for future) to have a window of size > 0
)