diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go index 157f6bc..01a4fcb 100644 --- a/sdks/scheduler/models.go +++ b/sdks/scheduler/models.go @@ -70,6 +70,7 @@ var JobInfoTypeUnion = types.NewTypeUnion[JobInfo]( (*HPCJobInfo)(nil), (*BindingJobInfo)(nil), (*PCMInferenceJobInfo)(nil), + (*FinishJobInfo)(nil), ) var _ = serder.UseTypeUnionInternallyTagged(&JobInfoTypeUnion, "type") @@ -148,6 +149,12 @@ type AIJobInfo struct { ResourceChoice ResourceChoice `json:"resourceChoice"` } +type FinishJobInfo struct { + serder.Metadata `union:"finish"` + JobInfoBase + Type string `json:"type"` +} + type ResourceChoice struct { Type string `json:"type"` ResourceScopes []ResourceScope `json:"resourceScopes"`