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))