proper logging with loki

central_processing.py

@@ -3,12 +3,15 @@
 import numpy as np
 from beartype.typing import Dict, Optional, Any, NoReturn
 from loguru import logger
+import sys
 
 from utils.helpers import timer
 from config import config
 from neotemplate.base_central_processing import CPNeoTemplate
 
 
+logger.add(sys.stdout, format=" level={level} | {time:YYYY-MM-DD at HH:mm:ss} | {message}", level="INFO")
+
 class CentralProcessing(CPNeoTemplate):
     """
     Central processing unit for preprocessing, postprocessing, predicting and training.

neotemplate/base_central_processing.py

@@ -6,9 +6,12 @@
 import yaml
 import toml
 import requests
+import sys
 
 from config import config as cfg
 
+logger.add(sys.stdout, format=" level={level} | {time:YYYY-MM-DD at HH:mm:ss} | {message}", level="INFO")
+
 
 class CPNeoTemplate(nn.Module):
     """Central processing unit for the NeoTemplate."""

utils/helpers.py

@@ -3,6 +3,9 @@
 from functools import wraps
 from time import time
 from loguru import logger
+import sys
+
+logger.add(sys.stdout, format=" level={level} | {time:YYYY-MM-DD at HH:mm:ss} | {message}", level="INFO")
 
 
 def timer(orig_func: Callable):