debugpy ports in dockerfile - remove

livepeer · Aug 2, 2024 · a07620e · a07620e
1 parent fb5da84
commit a07620e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/worker/docker.go b/worker/docker.go
@@ -19,7 +19,7 @@ import (
 
 const containerModelDir = "/models"
 const containerPort = "8000/tcp"
-const pollingInterval = 500 * time.Millisecond
+const pollingInterval = 5000 * time.Millisecond
 const containerTimeout = 2 * time.Minute
 const externalContainerTimeout = 2 * time.Minute
 const optFlagsContainerTimeout = 5 * time.Minute
@@ -177,6 +177,7 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 		},
 		ExposedPorts: nat.PortSet{
 			containerPort: struct{}{},
+			"5678/tcp":    struct{}{},
 		},
 		Labels: map[string]string{
 			containerCreatorLabel: containerCreator,
@@ -205,6 +206,12 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 					HostPort: containerHostPort,
 				},
 			},
+			"5678/tcp": []nat.PortBinding{
+				{
+					HostIP:   "0.0.0.0",
+					HostPort: "5678",
+				},
+			},
 		},
 	}