Format file

tests/cameras/test_cameras.py

@@ -37,10 +37,7 @@ pytest -sx 'tests/test_cameras.py::test_camera[intelrealsense-True]'
 import numpy as np
 import pytest
 
-from lerobot.common.robot_devices.utils import (
-    RobotDeviceAlreadyConnectedError,
-    RobotDeviceNotConnectedError,
-)
+from lerobot.common.robot_devices.utils import RobotDeviceAlreadyConnectedError, RobotDeviceNotConnectedError
 from tests.utils import TEST_CAMERA_TYPES, make_camera, require_camera
 
 # Maximum absolute difference between two consecutive images recorded by a camera.
@@ -115,11 +112,7 @@ def test_camera(request, camera_type, mock):
     )
     # TODO(rcadene): properly set `rtol`
     np.testing.assert_allclose(
-        color_image,
-        async_color_image,
-        rtol=1e-5,
-        atol=MAX_PIXEL_DIFFERENCE,
-        err_msg=error_msg,
+        color_image, async_color_image, rtol=1e-5, atol=MAX_PIXEL_DIFFERENCE, err_msg=error_msg
     )
 
     # Test disconnecting
@@ -138,11 +131,7 @@ def test_camera(request, camera_type, mock):
     assert camera.color_mode == "bgr"
     bgr_color_image = camera.read()
     np.testing.assert_allclose(
-        color_image,
-        bgr_color_image[:, :, [2, 1, 0]],
-        rtol=1e-5,
-        atol=MAX_PIXEL_DIFFERENCE,
-        err_msg=error_msg,
+        color_image, bgr_color_image[:, :, [2, 1, 0]], rtol=1e-5, atol=MAX_PIXEL_DIFFERENCE, err_msg=error_msg
     )
     del camera
 
@@ -177,11 +166,7 @@ def test_camera(request, camera_type, mock):
         rot_color_image = camera.read()
 
         np.testing.assert_allclose(
-            rot_color_image,
-            manual_rot_img,
-            rtol=1e-5,
-            atol=MAX_PIXEL_DIFFERENCE,
-            err_msg=error_msg,
+            rot_color_image, manual_rot_img, rtol=1e-5, atol=MAX_PIXEL_DIFFERENCE, err_msg=error_msg
         )
         del camera
 
@@ -215,9 +200,7 @@ def test_save_images_from_cameras(tmp_path, request, camera_type, mock):
     if camera_type == "opencv":
         from lerobot.common.robot_devices.cameras.opencv import save_images_from_cameras
     elif camera_type == "intelrealsense":
-        from lerobot.common.robot_devices.cameras.intelrealsense import (
-            save_images_from_cameras,
-        )
+        from lerobot.common.robot_devices.cameras.intelrealsense import save_images_from_cameras
 
     # Small `record_time_s` to speedup unit tests
     save_images_from_cameras(tmp_path, record_time_s=0.02, mock=mock)
@@ -226,13 +209,7 @@ def test_save_images_from_cameras(tmp_path, request, camera_type, mock):
 @pytest.mark.parametrize("camera_type, mock", TEST_CAMERA_TYPES)
 @require_camera
 def test_camera_rotation(request, camera_type, mock):
-    config_kwargs = {
-        "camera_type": camera_type,
-        "mock": mock,
-        "width": 640,
-        "height": 480,
-        "fps": 30,
-    }
+    config_kwargs = {"camera_type": camera_type, "mock": mock, "width": 640, "height": 480, "fps": 30}
 
     # No rotation.
     camera = make_camera(**config_kwargs, rotation=None)