Fix: set freejoint of environment props

JeanElsner · Nov 20, 2023 · 358823c · 358823c
1 parent 546ada9
commit 358823c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dm_robotics/panda/environment.py b/src/dm_robotics/panda/environment.py
@@ -98,7 +98,8 @@ def build_task_environment(self) -> subtask_env.SubTaskEnvironment:
   def add_props(self, props: Sequence[prop.Prop]):
     self._props.extend(props)
     for p in self._props:
-      self._arena.add_free_entity(p)
+      frame = self._arena.add_free_entity(p)
+      p.set_freejoint(frame.freejoint)
 
   def add_entity_initializers(
       self,