hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

rename color to color_mode

Browse Source
This commit is contained in:
Remi Cadene
2024-07-10 19:54:23 +02:00
parent 95de4b7454
commit 1bc13f39bf
2 changed files with 19 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_cameras.py
View File

@@ -84,9 +84,9 @@ def test_camera(request):
del camera
# Test acquiring a bgr image
camera = OpenCVCamera(CAMERA_INDEX, color="bgr")
camera = OpenCVCamera(CAMERA_INDEX, color_mode="bgr")
camera.connect()
assert camera.color == "bgr"
assert camera.color_mode == "bgr"
bgr_color_image = camera.read()
assert np.allclose(color_image, bgr_color_image[:, :, [2, 1, 0]], rtol=1e-5, atol=MAX_PIXEL_DIFFERENCE)
del camera