diff --git a/worker/docker.go b/worker/docker.go index 8d7f97e0..529f5c2c 100644 --- 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", + }, + }, }, }