Browse Source

优化创建package代码

pull/52/head
JeshuaRen 8 months ago
parent
commit
a936cffa13
1 changed files with 12 additions and 3 deletions
  1. +12
    -3
      sdks/scheduler/models.go

+ 12
- 3
sdks/scheduler/models.go View File

@@ -126,8 +126,9 @@ var JobResourceTypeUnion = types.NewTypeUnion[JobResource](
(*NPU)(nil),
(*MLU)(nil),
(*DCU)(nil),
(*Memory)(nil),
(*MEMORY)(nil),
(*PRICE)(nil),
(*STORAGE)(nil),
)

var _ = serder.UseTypeUnionInternallyTagged(&JobResourceTypeUnion, "type")
@@ -144,6 +145,14 @@ type CPU struct {
Number int64 `json:"number"`
}

type STORAGE struct {
serder.Metadata `union:"STORAGE"`
JobResourceBase
Type string `json:"type"`
Name string `json:"name"`
Number int64 `json:"number"`
}

type GPU struct {
serder.Metadata `union:"GPU"`
JobResourceBase
@@ -160,8 +169,8 @@ type NPU struct {
Number int64 `json:"number"`
}

type Memory struct {
serder.Metadata `union:"Memory"`
type MEMORY struct {
serder.Metadata `union:"MEMORY"`
JobResourceBase
Type string `json:"type"`
Name string `json:"name"`


Loading…
Cancel
Save