From 09aedeea0234226c967d442f5a2abb6caeecbc17 Mon Sep 17 00:00:00 2001 From: JamesWrigley Date: Sun, 17 Mar 2024 11:55:17 +0100 Subject: [PATCH] Special-case StreamingFunction in EagerThunkMetadata() constructor This always us to handle all the other kinds of task specs. --- src/submission.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/submission.jl b/src/submission.jl index 9627c77a3..8d5bc8473 100644 --- a/src/submission.jl +++ b/src/submission.jl @@ -184,7 +184,7 @@ function eager_process_options_submission_to_local(id_map, options::NamedTuple) end end function EagerThunkMetadata(spec::EagerTaskSpec) - f = chunktype(spec.f).instance + f = spec.f isa StreamingFunction ? spec.f.f : spec.f arg_types = ntuple(i->chunktype(spec.args[i][2]), length(spec.args)) return_type = Base.promote_op(f, arg_types...) return EagerThunkMetadata(return_type)