From 81f17d505ef6a7fd40c8ad86cd3c20cd7408d812 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Fri, 27 Sep 2024 12:21:06 +0200 Subject: [PATCH] if not '~cameras' in overrides --- tests/utils.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/tests/utils.py b/tests/utils.py index 51356edf8..b013a6b41 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -277,26 +277,29 @@ def make_robot(robot_type: str, overrides: list[str] | None = None, mock=False) if robot_type == "koch": overrides.append("+leader_arms.main.mock=true") overrides.append("+follower_arms.main.mock=true") - overrides.append("+cameras.laptop.mock=true") - overrides.append("+cameras.phone.mock=true") + if "~cameras" not in overrides: + overrides.append("+cameras.laptop.mock=true") + overrides.append("+cameras.phone.mock=true") elif robot_type == "koch_bimanual": overrides.append("+leader_arms.left.mock=true") overrides.append("+leader_arms.right.mock=true") overrides.append("+follower_arms.left.mock=true") overrides.append("+follower_arms.right.mock=true") - overrides.append("+cameras.laptop.mock=true") - overrides.append("+cameras.phone.mock=true") + if "~cameras" not in overrides: + overrides.append("+cameras.laptop.mock=true") + overrides.append("+cameras.phone.mock=true") elif robot_type == "aloha": overrides.append("+leader_arms.left.mock=true") overrides.append("+leader_arms.right.mock=true") overrides.append("+follower_arms.left.mock=true") overrides.append("+follower_arms.right.mock=true") - overrides.append("+cameras.cam_high.mock=true") - overrides.append("+cameras.cam_low.mock=true") - overrides.append("+cameras.cam_left_wrist.mock=true") - overrides.append("+cameras.cam_right_wrist.mock=true") + if "~cameras" not in overrides: + overrides.append("+cameras.cam_high.mock=true") + overrides.append("+cameras.cam_low.mock=true") + overrides.append("+cameras.cam_left_wrist.mock=true") + overrides.append("+cameras.cam_right_wrist.mock=true") else: raise NotImplementedError(robot_type)