admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-19 18:49:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

formatting

Browse Source
This commit is contained in:
Pepijn
2026-02-13 17:46:18 +01:00
parent 1de2b87a92
commit 4fa41ba806
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_delta_actions.py
+7 -5
View File
@@ -25,11 +25,13 @@ def dataset(tmp_path, empty_lerobot_dataset_factory):
ds = empty_lerobot_dataset_factory(root=tmp_path / "delta_test", features=features) ds = empty_lerobot_dataset_factory(root=tmp_path / "delta_test", features=features)
for ep in range(2): for ep in range(2):
for _ in range(5): for _ in range(5):
ds.add_frame({ ds.add_frame(
"action": np.random.randn(ACTION_DIM).astype(np.float32), {
"observation.state": np.random.randn(STATE_DIM).astype(np.float32), "action": np.random.randn(ACTION_DIM).astype(np.float32),
"task": f"task_{ep}", "observation.state": np.random.randn(STATE_DIM).astype(np.float32),
}) "task": f"task_{ep}",
}
)
ds.save_episode() ds.save_episode()
ds.finalize() ds.finalize()
return ds return ds