diff --git a/central_processing.py b/central_processing.py index 9c3e06d..acf79ae 100755 --- a/central_processing.py +++ b/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. diff --git a/neotemplate/base_central_processing.py b/neotemplate/base_central_processing.py index 11514a0..24d51b3 100644 --- a/neotemplate/base_central_processing.py +++ b/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.""" diff --git a/utils/helpers.py b/utils/helpers.py index 0b3c398..8171fe6 100644 --- a/utils/helpers.py +++ b/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):