fix: wrap python basic logs in debug assertion in launcher (#2539)

* fix: wrap python basic logs in debug assertion in launcher * use level filters instead
huggingface · Sep 20, 2024 · 7e2d188 · 7e2d188
1 parent f478aa7
commit 7e2d188
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/launcher/src/main.rs b/launcher/src/main.rs
@@ -1038,11 +1038,13 @@ fn log_lines<R: Sized + Read>(mut bufread: BufReader<R>) {
                         Ok(log) => log.trace(),
                         // For interactive debugging ?
                         Err(_) => {
-                            stdout.write_all(line).unwrap();
-                            if lines.peek().is_some() {
-                                stdout.write_all(b"\n").unwrap();
+                            if LevelFilter::current() >= tracing::Level::DEBUG {
+                                stdout.write_all(line).unwrap();
+                                if lines.peek().is_some() {
+                                    stdout.write_all(b"\n").unwrap();
+                                }
+                                stdout.flush().unwrap();
                             }
-                            stdout.flush().unwrap();
                         }
                     }
                 }