diff --git a/sdks/pcmscheduler/models.go b/sdks/pcmscheduler/models.go index 7f2338b..a116ab8 100644 --- a/sdks/pcmscheduler/models.go +++ b/sdks/pcmscheduler/models.go @@ -102,7 +102,8 @@ const ( Inference = "Inference" Train = "Train" - BroadcastWait = "All" + BroadcastWait = "BroadcastWait" + WaitBroadcast = "WaitBroadcast" Broadcast = "Broadcast" Wait = "Wait" ) diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go index 2a7b0ef..1a2c537 100644 --- a/sdks/scheduler/models.go +++ b/sdks/scheduler/models.go @@ -188,19 +188,19 @@ type UploadJobInfo struct { type BroadcastWaitInfo struct { serder.Metadata `union:"BroadcastWait"` JobInfoBase - Type string `json:"type"` - HandleType string `json:"handleType"` - Broadcast Broadcast `json:"broadcast"` - Wait Wait `json:"wait"` + Type string `json:"type"` + HandleType string `json:"handleType"` + Broadcast []Broadcast `json:"broadcast"` + Wait []Wait `json:"wait"` } type Broadcast struct { NextJobSetID JobSetID `json:"nextJobSetID"` - NextLocalJobID JobID `json:"nextLocalJobID"` + NextLocalJobID string `json:"nextLocalJobID"` } type Wait struct { - WaitJobIDs []JobID `json:"waitJobIDs"` + WaitJobID string `json:"waitJobID"` } type NotifyJobInfo struct {