Improve logging in main.py

RiverBench · Sep 10, 2024 · cda7fb9 · cda7fb9
1 parent 22a7e17
commit cda7fb9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -1,12 +1,15 @@
 # See: https://mkdocs-macros-plugin.readthedocs.io/en/latest
 
-import os
+import os, logging
+
+
+log = logging.getLogger('mkdocs')
 
 
 def define_env(env):
     tag = os.environ.get('RELEASE_TAG', '')
     if not tag:
-        print('Warning: RELEASE_TAG environment variable not set. Defaulting to "dev".')
+        log.warning('RELEASE_TAG environment variable not set. Defaulting to "dev".')
         tag = 'dev'