Merge pull request #486 from JamesWrigley/test-fixes

Special-case StreamingFunction in EagerThunkMetadata() constructor
JuliaParallel · Mar 20, 2024 · 29d790f · 29d790f
2 parents b4f41c8 + 09aedee
commit 29d790f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)