robosuite_task_zoo/environments/manipulation/hammer_place.py
def _setup_references(self):
        """
        Sets up references to important components. A reference is typically an
        index or a list of indices that point to the corresponding elements
        in a flatten array, which is how MuJoCo stores physical simulation data.
        """
        super()._setup_references()

        # Additional object references from this env
        self.object_body_ids = dict()

        self.cabinet_qpos_addrs = self.sim.model.get_joint_qpos_addr(self.cabinet_object.joints[0])

        self.sorting_object_id = self.sim.model.body_name2id(self.sorting_object.root_body)
        self.cabinet_object_id = self.sim.model.body_name2id(self.cabinet_object.root_body)
        
        self.obj_body_id = {}        
        for obj in self.objects:
            self.obj_body_id[obj.name] = self.sim.model.body_name2id(obj.root_body)

        Sets up references to important components. A reference is typically an
        index or a list of indices that point to the corresponding elements
        in a flatten array, which is how MuJoCo stores physical simulation data.
        
Sets up references to important components. A reference is typically an
index or a list of indices that point to the corresponding elements
in a flatten array, which is how MuJoCo stores physical simulation data.
Sets up references to important components . A reference is typically an index or a list of indices that point to the corresponding elements in a flatten array which is how MuJoCo stores physical simulation data .
def _setup_references(self):
        super()._setup_references()
        self.object_body_ids = dict()
        self.cabinet_qpos_addrs = self.sim.model.get_joint_qpos_addr(self.cabinet_object.joints[0])
        self.sorting_object_id = self.sim.model.body_name2id(self.sorting_object.root_body)
        self.cabinet_object_id = self.sim.model.body_name2id(self.cabinet_object.root_body)
        self.obj_body_id = {}        
        for obj in self.objects:
            self.obj_body_id[obj.name] = self.sim.model.body_name2id(obj.root_body)
def _setup_references ( self ) : super ( ) . _setup_references ( ) self . object_body_ids = dict ( ) self . cabinet_qpos_addrs = self . sim . model . get_joint_qpos_addr ( self . cabinet_object . joints [ 0 ] ) self . sorting_object_id = self . sim . model . body_name2id ( self . sorting_object . root_body ) self . cabinet_object_id = self . sim . model . body_name2id ( self . cabinet_object . root_body ) self . obj_body_id = { } for obj in self . objects : self . obj_body_id [ obj . name ] = self . sim . model . body_name2id ( obj . root_body )
Sets up references to important components.
Sets up references to important components .
"""
        Sets up references to important components. A reference is typically an
        index or a list of indices that point to the corresponding elements
        in a flatten array, which is how MuJoCo stores physical simulation data.
        """ # Additional object references from this env
self