chore: add env variable for debugging

kazewong · Sep 27, 2024 · 78a3555 · 78a3555
1 parent a674f85
commit 78a3555
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/flowMC/utils/debug.py b/src/flowMC/utils/debug.py
@@ -1,15 +1,19 @@
 # Copyright (c) 2022 Kaze Wong & contributor
 
 
+import os
+
 import jax.debug
 
 DEBUG_VERBOSE = False
 
 
-def enable_debug_verbose() -> None:
+def enable_debug_verbose(verbose=True) -> None:
     r"""Enables verbose in debugging mode."""
+    if not verbose:
+        verbose = os.getenv("FLOWMC_DEBUG_VERBOSE", 0)
     global DEBUG_VERBOSE
-    DEBUG_VERBOSE = True
+    DEBUG_VERBOSE = bool(verbose)
 
 
 def get_mode() -> bool: