Merge pull request #103 from tue-robotics/fix/hero_in_ac2

Added Hero as robot object
tue-robotics · Jan 15, 2019 · 0749c69 · 0749c69
2 parents de3e2fd + 2f89f78
commit 0749c69
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/action_server/src/action_server/main.py b/action_server/src/action_server/main.py
@@ -11,7 +11,7 @@
     rospy.init_node('action_server')
 
     try:
-        robot_name = rospy.get_param('~robot_name')
+        robot_name = rospy.get_param('~robot_name').lower()
     except KeyError:
         rospy.logerr("Please provide a 'robot_name'")
         exit(0)
@@ -23,8 +23,10 @@
         from robot_skills.amigo import Amigo as Robot
     elif robot_name == 'sergio':
         from robot_skills.sergio import Sergio as Robot
+    elif robot_name == 'hero':
+        from robot_skills.hero import Hero as Robot
     else:
-        rospy.logerr("'robot_name' must be either 'amigo' or 'sergio'")
+        rospy.logerr("'robot_name' must be 'amigo', 'sergio' or 'hero'")
         sys.exit()
 
     robot = Robot()