admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-26 03:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(rollout): go back to initial position is now a config

Browse Source
This commit is contained in:
Steven Palma
2026-04-27 16:09:30 +02:00
parent bef67960d0
commit 7451813da6
7 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_rollout.py
+2 -2
View File
@@ -254,7 +254,7 @@ def test_create_inference_engine_sync():
def test_estimate_max_episode_seconds_no_video():
from lerobot.rollout.strategies import estimate_max_episode_seconds
assert estimate_max_episode_seconds({}, fps=30.0) == 600.0
assert estimate_max_episode_seconds({}, fps=30.0) == 300.0
def test_estimate_max_episode_seconds_with_video():
@@ -264,7 +264,7 @@ def test_estimate_max_episode_seconds_with_video():
result = estimate_max_episode_seconds(features, fps=30.0)
assert result > 0
# With a real camera, duration should differ from the fallback
assert result != 600.0
assert result != 300.0
def test_safe_push_to_hub():