fix: setup input keys correctly.

LASR-at-Home · Apr 18, 2024 · df1db0c · df1db0c
1 parent a94b357
commit df1db0c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 110 deletions.
diff --git a/skills/src/lasr_skills/find_known_person.py b/skills/src/lasr_skills/find_known_person.py
diff --git a/skills/src/lasr_skills/go_to_location.py b/skills/src/lasr_skills/go_to_location.py
@@ -26,7 +26,13 @@
 class GoToLocation(smach.StateMachine):
 
     def __init__(self, location: Union[Pose, None] = None):
-        super(GoToLocation, self).__init__(outcomes=["succeeded", "failed"])
+
+        if location is not None:
+            super(GoToLocation, self).__init__(outcomes=["succeeded", "failed"])
+        else:
+            super(GoToLocation, self).__init__(
+                outcomes=["succeeded", "failed"], input_keys=["location"]
+            )
 
         IS_SIMULATION = (
             "/pal_startup_control/start" not in rosservice.get_service_list()