Added a helper method to extract single jobs #16

Merged
schmittlauch merged 1 commit from mapping_problem_helper into mapping_problem 2017-07-09 17:20:34 +02:00

View file

@ -93,7 +93,7 @@ class JobShopProblem(Mapping):
def __len__(self) -> int:
return sum(map(len, self.problem_data))
def get_tasks_by_job(self, job):
def get_tasks_by_job(self, job: int) -> List[Tuple[int, int]]:
return self.problem_data[job]
class JobShopVisitor(arpeggio.PTNodeVisitor):