fix: do not expose ports for custom image (#754)

Signed-off-by: Keming <kemingyang@tensorchord.ai> Signed-off-by: Keming <kemingyang@tensorchord.ai>
tensorchord · Aug 11, 2022 · ad4b9ec · ad4b9ec
1 parent e77b7d6
commit ad4b9ec
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/lang/ir/compile.go b/pkg/lang/ir/compile.go
@@ -132,6 +132,12 @@ func (g Graph) Labels() (map[string]string, error) {
 
 func (g Graph) ExposedPorts() (map[string]struct{}, error) {
  ports := make(map[string]struct{})
+
+ // do not expose ports for custom images
+ if g.Image != nil {
+ return ports, nil
+ }
+
  ports[fmt.Sprintf("%d/tcp", config.SSHPortInContainer)] = struct{}{}
  if g.JupyterConfig != nil {
  ports[fmt.Sprintf("%d/tcp", config.JupyterPortInContainer)] = struct{}{}