add MOCK_PIPELINE env var to runner if exists

livepeer · Jun 28, 2024 · 5cbb6cc · 5cbb6cc
1 parent a36337e
commit 5cbb6cc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/worker/docker.go b/worker/docker.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"log/slog"
+	"os"
 	"strings"
 	"sync"
 	"time"
@@ -164,6 +165,12 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
 		"PIPELINE=" + pipeline,
 		"MODEL_ID=" + modelID,
 	}
+	_, ok := os.LookupEnv("MOCK_PIPELINE")
+	if ok {
+		envVars = append(envVars, "MOCK_PIPELINE=true")
+		slog.Info("MOCK_PIPELINE set, passing to runner container")
+	}
+
 	for key, value := range optimizationFlags {
 		envVars = append(envVars, key+"="+value.String())
 	}