forked from kubeflow/arena
-
Notifications
You must be signed in to change notification settings - Fork 0
/
model.go
33 lines (28 loc) · 953 Bytes
/
model.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
package model
import "github.com/spf13/cobra"
var (
serveLong = `submit a model analyze job.
Available Commands:
profile Submit a model profile job.
evaluate Submit a model evaluate job.
optimize Submit a model optimize job.
benchmark Submit a model benchmark job`
)
func NewModelCommand() *cobra.Command {
var command = &cobra.Command{
Use: "model",
Short: "Submit a model analyze job. (experimental feature)",
Long: serveLong,
Run: func(cmd *cobra.Command, args []string) {
cmd.HelpFunc()(cmd, args)
},
}
command.AddCommand(NewSubmitModelProfileJobCommand())
command.AddCommand(NewSubmitModelOptimizeJobCommand())
command.AddCommand(NewSubmitModelBenchmarkJobCommand())
command.AddCommand(NewSubmitModelEvaluateJobCommand())
command.AddCommand(NewGetModelJobCommand())
command.AddCommand(NewListModelJobsCommand())
command.AddCommand(NewDeleteModelJobCommand())
return command
}