Make log level compilation flag (#109)

- debug build will have INFO as default log level - release build will have WARN as default log level
huggingface · Dec 9, 2024 · 43b6bae · 43b6bae
1 parent 2a9af04
commit 43b6bae
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/hf_xet/src/log.rs b/hf_xet/src/log.rs
@@ -9,6 +9,11 @@ use utils::ThreadPool;
 
 use crate::log_buffer::{get_telemetry_task, LogBufferLayer, TELEMETRY_PRE_ALLOC_BYTES};
 
+/// Default log level for the library to use. Override using `RUST_LOG` env variable.
+#[cfg(not(debug_assertions))]
+const DEFAULT_LOG_LEVEL: &str = "warn";
+
+#[cfg(debug_assertions)]
 const DEFAULT_LOG_LEVEL: &str = "info";
 
 pub fn initialize_logging(threadpool: Arc<ThreadPool>) {