From 481d60dc848502d51bd66fca1e40a705786713c1 Mon Sep 17 00:00:00 2001 From: Kevin Lin Date: Wed, 26 Feb 2025 20:41:08 -0600 Subject: [PATCH] Improve logging if invalid object to scale --- robosuite/models/arenas/arena.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/robosuite/models/arenas/arena.py b/robosuite/models/arenas/arena.py index 7d9629936..615317e9d 100644 --- a/robosuite/models/arenas/arena.py +++ b/robosuite/models/arenas/arena.py @@ -184,7 +184,9 @@ def set_scale(self, scale: Union[float, List[float]], obj_name: str): """ obj = self.worldbody.find(f"./body[@name='{obj_name}']") if obj is None: - raise ValueError(f"Object {obj_name} not found in arena; cannot set scale.") + bodies = self.worldbody.findall("./body") + body_names = [body.get("name") for body in bodies if body.get("name") is not None] + raise ValueError(f"Object {obj_name} not found in arena; cannot set scale. Available objects: {body_names}") self.object_scales[obj.get("name")] = scale # scale geoms