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

rename action_horizon to chunk_size

Browse Source
This commit is contained in:
Pepijn
2025-09-11 19:42:25 +02:00
parent b044f3104b
commit 2234b851c0
8 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_pi0_openpi.py
+1 -1
View File
@@ -36,7 +36,7 @@ def test_policy_instantiation():
device = policy.device if hasattr(policy, "device") else "cpu"
batch = {
"observation.state": torch.randn(batch_size, 14, dtype=torch.float32, device=device),
"action": torch.randn(batch_size, config.action_horizon, 7, dtype=torch.float32, device=device),
"action": torch.randn(batch_size, config.chunk_size, 7, dtype=torch.float32, device=device),
"observation.images.base_0_rgb": torch.rand(
batch_size, 3, 224, 224, dtype=torch.float32, device=device
), # Use rand for [0,1] range