chore: skip starship prompt for customized image (#687)

Signed-off-by: Keming <[email protected]>
tensorchord · Jul 29, 2022 · df9bd6a · df9bd6a
1 parent 78cba5f
commit df9bd6a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/lang/ir/shell.go b/pkg/lang/ir/shell.go
@@ -51,6 +51,10 @@ func (g *Graph) compileShell(root llb.State) (llb.State, error) {
 }
 
 func (g *Graph) compilePrompt(root llb.State) llb.State {
+	// skip this for customized image
+	if g.Image != nil {
+		return root
+	}
 	// starship config
 	config := root.
 		File(llb.Mkdir(defaultConfigDir, 0755, llb.WithParents(true)),