Add robot check also to room filter

sea-bass · Dec 22, 2024 · 1958b1c · 1958b1c
1 parent 903bbe4
commit 1958b1c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pyrobosim/pyrobosim/utils/knowledge.py b/pyrobosim/pyrobosim/utils/knowledge.py
@@ -332,15 +332,19 @@ def resolve_to_object(
         possible_objects = [
             obj
             for obj in possible_objects
-            if obj.parent.parent.parent.name == room_name
+            if (
+                # Check whether the parent is a robot before performing further checks
+                hasattr(obj.parent, "parent")
+                and obj.parent.parent.parent.name == room_name
+            )
         ]
 
     if location is not None:
         possible_objects = [
             obj
             for obj in possible_objects
             if (
-                # Check whether the parent is a robot performing further checks
+                # Check whether the parent is a robot before performing further checks
                 hasattr(obj.parent, "parent")
                 and (
                     obj.parent == location