use env var to set shared logging level when sampling

henryaddison · Feb 28, 2024 · 03ebfec · 03ebfec
1 parent 6e7ef98
commit 03ebfec
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/bin/predict.py b/bin/predict.py
@@ -56,11 +56,10 @@
 load_dotenv()  # take environment variables from .env.
 
 logging.basicConfig(
-    level=logging.INFO,
+    level=os.environ.get("LOG_LEVEL", "INFO").upper(),
     format="%(levelname)s - %(filename)s - %(asctime)s - %(message)s",
 )
 logger = logging.getLogger()
-logger.setLevel(os.environ.get("LOG_LEVEL", "INFO").upper())
 
 app = typer.Typer()