feat(cli): Support command flag in deploy (#95)

Signed-off-by: Ce Gao <[email protected]>
tensorchord · Aug 2, 2023 · 0aeb6c0 · 0aeb6c0
1 parent 943761c
commit 0aeb6c0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/mdz/README.md b/mdz/README.md
@@ -13,7 +13,9 @@ CLI for OpenModelZ.
 
 ## Installation
 
-TODO
+```
+pip install openmodelz
+```
 
 ## CLI Reference
 

diff --git a/mdz/pkg/cmd/deploy.go b/mdz/pkg/cmd/deploy.go
@@ -20,6 +20,7 @@ var (
 	deployName        string
 	deployGPU         int
 	deployNodeLabel   []string
+	deployCommand     string
 )
 
 // deployCmd represents the deploy command
@@ -53,6 +54,7 @@ func init() {
 	deployCmd.Flags().IntVar(&deployGPU, "gpu", 0, "Number of GPUs")
 	deployCmd.Flags().StringVar(&deployName, "name", "", "Name of inference")
 	deployCmd.Flags().StringSliceVarP(&deployNodeLabel, "node-labels", "l", []string{}, "Node labels")
+	deployCmd.Flags().StringVar(&deployCommand, "command", "", "Command to run")
 }
 
 func commandDeploy(cmd *cobra.Command, args []string) error {
@@ -87,6 +89,10 @@ func commandDeploy(cmd *cobra.Command, args []string) error {
 		},
 	}
 
+	if deployCommand != "" {
+		inf.Spec.Command = &deployCommand
+	}
+
 	if len(deployNodeLabel) > 0 {
 		inf.Spec.Constraints = []string{}
 		for _, label := range deployNodeLabel {
-Original file line number
+Diff line change
@@ Expand Up / @@ -13,7 +13,9 @@ CLI for OpenModelZ. @@
     ## Installation
-    TODO
+    ```
+    pip install openmodelz
+    ```
     ## CLI Reference
@@ Expand Down @@