Fix unit tests

This commit is contained in:
Remi Cadene
2024-09-25 14:11:28 +02:00
parent 6377d2a96c
commit bded8cbbe9
4 changed files with 31 additions and 10 deletions

View File

@@ -204,7 +204,7 @@ def require_robot(func):
# Run test with a monkeypatched version of the robot devices.
if mock:
# TODO(rcadene): redesign mocking to not have this hardcoded logic
if robot_type == "koch":
if robot_type in ["koch", "koch_bimanual"]:
camera_type = "opencv"
elif robot_type == "aloha":
camera_type = "intelrealsense"