def set_margin()

in gym-compete/gym_compete/new_envs/agents/agent.py [0:0]

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

    def set_margin(self, margin):
        agent_geom_ids = [i for i, name in enumerate(self.env.model.geom_names)
                          if self.in_scope(name)]
        m = self.env.model.geom_margin.copy()
        print("Resetting", self.scope, "margins to", margin)
        m[agent_geom_ids] = margin
        self.env.model.__setattr__('geom_margin', m)