format

tests/test_control_robot.py

@@ -1,6 +1,5 @@
-
-
 from pathlib import Path
+
 from lerobot.common.policies.factory import make_policy
 from lerobot.common.robot_devices.robots.factory import make_robot
 from lerobot.common.utils.utils import init_hydra_config
@@ -21,12 +20,14 @@ def test_record_dataset_and_replay_episode_and_run_policy(tmpdir):
     env_name = "koch_real"
     policy_name = "act_koch_real"
 
-    #root = Path(tmpdir)
+    # root = Path(tmpdir)
     root = Path("tmp/data")
     repo_id = "lerobot/debug"
 
     robot = make_robot(robot_name)
-    dataset = record_dataset(robot, fps=30, root=root, repo_id=repo_id, warmup_time_s=1, episode_time_s=1, num_episodes=2)
+    dataset = record_dataset(
+        robot, fps=30, root=root, repo_id=repo_id, warmup_time_s=1, episode_time_s=1, num_episodes=2
+    )
 
     replay_episode(robot, episode=0, fps=30, root=root, repo_id=repo_id)
 
@@ -36,7 +37,7 @@ def test_record_dataset_and_replay_episode_and_run_policy(tmpdir):
             f"env={env_name}",
             f"policy={policy_name}",
             f"device={DEVICE}",
-        ]
+        ],
     )
 
     policy = make_policy(hydra_cfg=cfg, dataset_stats=dataset.stats)
@@ -44,4 +45,3 @@ def test_record_dataset_and_replay_episode_and_run_policy(tmpdir):
     run_policy(robot, policy, cfg, run_time_s=1)
 
     del robot
-