Fix error return of the pod log writer

trustyai-explainability · Nov 11, 2024 · 4cfaac9 · 4cfaac9
1 parent e648080
commit 4cfaac9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/controllers/lmes/driver/driver.go b/controllers/lmes/driver/driver.go
@@ -290,7 +290,7 @@ type PodLogAndPipeWriter struct {
 func (f PodLogAndPipeWriter) Write(p []byte) (n int, err error) {
 	// format lm-eval output to stdout and the pipewriter
 	line := string(p)
-	fmt.Print(line)
+	res, _ := fmt.Print(line)
 	if _, err := f.pw.Write(p); err != nil {
 		return 0, err
 	}
@@ -303,7 +303,7 @@ func (f PodLogAndPipeWriter) Write(p []byte) (n int, err error) {
 			return 0, err
 		}
 	}
-	return 0, nil
+	return res, nil
 }
 
 func (d *driverImpl) exec() error {