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

Signed-off-by: Keming <kemingyang@tensorchord.ai>
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)),