def get_allocation(self, assignment):
allocation = Allocation([])
for i_e, e in enumerate(assignment):
r_id = self.robots[e].id
a_id = self.tasks[i_e].id
allocation.add(r_id, a_id)
return allocation
def get_allocation(self):
allocation = Allocation([])
for r in self.robots:
for a in r.S_r:
allocation.add(r.id, a.id)
return allocation
