def populate_work_queue(self) -> None:
self.work_queue = []
# Only populate the work queue summaries if the pool is initialized. We
# can then be sure that the queue is available in the operations below.
if self.state == PoolState.init:
return
worksets = peek_queue(self.get_pool_queue(),
StorageType.corpus,
object_type=WorkSet)
for workset in worksets:
work_units = [
WorkUnitSummary(
job_id=work_unit.job_id,
task_id=work_unit.task_id,
task_type=work_unit.task_type,
) for work_unit in workset.work_units
]
self.work_queue.append(WorkSetSummary(work_units=work_units))
def populate_work_queue(self) -> None:
self.work_queue = []
# Only populate the work queue summaries if the pool is initialized. We
# can then be sure that the queue is available in the operations below.
if self.state == PoolState.init:
return
worksets = self.peek_work_queue()
for workset in worksets:
# only include work units with work
if not workset.work_units:
continue
work_units = [
WorkUnitSummary(
job_id=work_unit.job_id,
task_id=work_unit.task_id,
task_type=work_unit.task_type,
) for work_unit in workset.work_units
]
self.work_queue.append(WorkSetSummary(work_units=work_units))