pub fn dispatch_custom_sample_aggregation(
plan: &ExecutionPlan,
controllers: &RuntimeControllerRegistry,
task_id: impl Into<String>,
block: PredictionBlock,
relations: SampleRelationSet,
policy: AggregationPolicy,
requested_unit_order: Vec<PredictionUnitId>,
) -> Result<AggregatedPredictionBlock, DagMlError>