Skip to content

Commit ff7c389

Browse files
authored
Fix an issue when running a pipeline with prism runner on docker in python. (#36002)
1 parent aca551a commit ff7c389

File tree

1 file changed

+9
-1
lines changed
  • sdks/go/pkg/beam/runners/prism/internal/worker

1 file changed

+9
-1
lines changed

sdks/go/pkg/beam/runners/prism/internal/worker/worker.go

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,14 @@ func (wk *W) GetProvisionInfo(_ context.Context, _ *fnpb.GetProvisionInfoRequest
156156
endpoint := &pipepb.ApiServiceDescriptor{
157157
Url: wk.Endpoint(),
158158
}
159+
160+
var rt string
161+
if len(wk.EnvPb.GetDependencies()) > 0 {
162+
rt = wk.JobKey
163+
} else {
164+
rt = "__no_artifacts_staged__"
165+
}
166+
159167
resp := &fnpb.GetProvisionInfoResponse{
160168
Info: &fnpb.ProvisionInfo{
161169
// TODO: Include runner capabilities with the per job configuration.
@@ -168,7 +176,7 @@ func (wk *W) GetProvisionInfo(_ context.Context, _ *fnpb.GetProvisionInfoRequest
168176
Url: wk.ArtifactEndpoint,
169177
},
170178

171-
RetrievalToken: wk.JobKey,
179+
RetrievalToken: rt,
172180
Dependencies: wk.EnvPb.GetDependencies(),
173181
PipelineOptions: wk.PipelineOptions,
174182

0 commit comments

Comments
 (0)