Skip to content
Snippets Groups Projects

Pipeline task management

Merged Christopher Randolph Rhodes requested to merge dev_task_queue into staging
3 files
+ 16
11
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -58,20 +58,14 @@ class RoiSetObjectMapParams(PipelineParams):
class RoiSetToObjectMapRecord(PipelineRecord):
pass
# TODO: shorten endpoint name, maybe simplify return typespec
@router.put('/roiset_to_obmap/infer')
def roiset_object_map(p: RoiSetObjectMapParams) -> RoiSetToObjectMapRecord:
def roiset_object_map(p: RoiSetObjectMapParams) -> Union[RoiSetToObjectMapRecord, PipelineQueueRecord]:
"""
Compute a RoiSet from 2d segmentation, apply to z-stack, and optionally apply object classification.
"""
return call_pipeline(roiset_object_map_pipeline, p)
@router.put('/queue/roiset_to_obmap')
def queue_roiset_object_map(p: RoiSetObjectMapParams) -> PipelineQueueRecord:
task_id = session.queue.add_task(
lambda x: call_pipeline(roiset_object_map_pipeline, x),
p
)
return {'task_id': task_id}
def roiset_object_map_pipeline(
accessors: Dict[str, GenericImageDataAccessor],
Loading