def update_layout()

in safety_gym/envs/engine.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def update_layout(self):
        ''' Update layout dictionary with new places of objects '''
        self.sim.forward()
        for k in list(self.layout.keys()):
            # Mocap objects have to be handled separately
            if 'gremlin' in k:
                continue
            self.layout[k] = self.data.get_body_xpos(k)[:2].copy()