Skip to content

Commit 0f747ed

Browse files
JamesWrigleyjpsamaroo
authored andcommitted
Special-case StreamingFunction in EagerThunkMetadata() constructor
This always us to handle all the other kinds of task specs.
1 parent 224d8ed commit 0f747ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: src/submission.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ function eager_process_options_submission_to_local(id_map, options::NamedTuple)
210210
end
211211
end
212212
function EagerThunkMetadata(spec::EagerTaskSpec)
213-
f = chunktype(spec.f).instance
213+
f = spec.f isa StreamingFunction ? spec.f.f : spec.f
214214
arg_types = ntuple(i->chunktype(spec.args[i][2]), length(spec.args))
215215
return_type = Base.promote_op(f, arg_types...)
216216
return EagerThunkMetadata(return_type)

0 commit comments

Comments
 (0)