Add neologger

central_processing.py

@@ -2,15 +2,14 @@
 from typing import Dict, Any
 import numpy as np
 from beartype.typing import Dict, Optional, Any, NoReturn
-from loguru import logger
-import sys
+from neolibrary.monitoring.logger import NeoLogger
 
 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")
+logger = NeoLogger(__name__).get_logger()
 
 class CentralProcessing(CPNeoTemplate):
     """

data_download.py

@@ -1,13 +1,15 @@
 from pymedquery import pymq
 import os
-from loguru import logger
 from tqdm import tqdm
 import h5py
+from neolibrary.monitoring.logger import NeoLogger
 import nibabel as nib
 from typing import NoReturn
 
 import config.config as cfg
 
+logger = NeoLogger(__name__).get_logger()
+
 
 class MedqueryDataDownloader:
     """

neotemplate/base_central_processing.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from loguru import logger
+from neolibrary.monitoring.logger import NeoLogger
 from beartype.typing import Dict, Optional, Any, NoReturn
 import numpy as np
 import yaml
@@ -10,7 +10,8 @@
 
 from config import config as cfg
 
-logger.add(sys.stdout, format=" level={level} | {time:YYYY-MM-DD at HH:mm:ss} | {message}", level="INFO")
+
+logger = NeoLogger(__name__).get_logger()
 
 
 class CPNeoTemplate(nn.Module):