Special-case StreamingFunction in EagerThunkMetadata() constructor

This always us to handle all the other kinds of task specs.
JuliaParallel · Apr 5, 2024 · fa01410 · fa01410
1 parent 3b90d08
commit fa01410
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/submission.jl b/src/submission.jl
@@ -218,7 +218,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)