From 4fa41ba8061ba716f85c23b6532ed5f71c81d65e Mon Sep 17 00:00:00 2001 From: Pepijn Date: Fri, 13 Feb 2026 17:46:18 +0100 Subject: [PATCH] formatting --- tests/policies/test_delta_actions.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tests/policies/test_delta_actions.py b/tests/policies/test_delta_actions.py index 0900123d8..f37bd6461 100644 --- a/tests/policies/test_delta_actions.py +++ b/tests/policies/test_delta_actions.py @@ -25,11 +25,13 @@ def dataset(tmp_path, empty_lerobot_dataset_factory): ds = empty_lerobot_dataset_factory(root=tmp_path / "delta_test", features=features) for ep in range(2): for _ in range(5): - ds.add_frame({ - "action": np.random.randn(ACTION_DIM).astype(np.float32), - "observation.state": np.random.randn(STATE_DIM).astype(np.float32), - "task": f"task_{ep}", - }) + ds.add_frame( + { + "action": np.random.randn(ACTION_DIM).astype(np.float32), + "observation.state": np.random.randn(STATE_DIM).astype(np.float32), + "task": f"task_{ep}", + } + ) ds.save_episode() ds.finalize() return ds