Merge pull request #66 from haosulab/fix_planner_init

Fix Planner init bug
haosulab · Feb 22, 2024 · 85f9de0 · 85f9de0
2 parents adbbba4 + 20e30a4
commit 85f9de0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/mplib/planner.py b/mplib/planner.py
@@ -72,6 +72,8 @@ def __init__(
             verbose=False,
         )
         self.pinocchio_model = self.robot.get_pinocchio_model()
+        self.user_link_names = self.pinocchio_model.get_link_names()
+        self.user_joint_names = self.pinocchio_model.get_joint_names()
 
         self.planning_world = PlanningWorld(
             [self.robot],
@@ -84,10 +86,6 @@ def __init__(
             self.generate_collision_pair()
             self.robot.update_SRDF(self.srdf)
 
-        self.pinocchio_model = self.robot.get_pinocchio_model()
-        self.user_link_names = self.pinocchio_model.get_link_names()
-        self.user_joint_names = self.pinocchio_model.get_joint_names()
-
         self.joint_name_2_idx = {}
         for i, joint in enumerate(self.user_joint_names):
             self.joint_name_2_idx[joint] = i