Skip to content

Commit fa01410

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 3b90d08 commit fa01410

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
@@ -218,7 +218,7 @@ function eager_process_options_submission_to_local(id_map, options::NamedTuple)
218218
end
219219
end
220220
function EagerThunkMetadata(spec::EagerTaskSpec)
221-
f = chunktype(spec.f).instance
221+
f = spec.f isa StreamingFunction ? spec.f.f : spec.f
222222
arg_types = ntuple(i->chunktype(spec.args[i][2]), length(spec.args))
223223
return_type = Base.promote_op(f, arg_types...)
224224
return EagerThunkMetadata(return_type)

0 commit comments

Comments
 (0)