f6584314b2
Add unit tests for connectors configuration, logger, workflows, and predictions batch - Implement tests for MLflow, OPC, and Postgres configuration builders to validate environment variable handling and default values. - Create tests for the logger to ensure default settings and handler configurations are correct. - Add comprehensive tests for the FormatAndExportPrediction and PredictionProcess workflows, covering various scenarios including path flags and activity execution. - Introduce tests for the PredictionsBatch workflow to verify the execution of local activities and child workflows. - Include a values.yaml file for Kubernetes deployment configuration, specifying image details, service account settings, environment variables, and resource limits.
23 lines
494 B
Python
23 lines
494 B
Python
from os import getenv
|
|
import logging
|
|
import sys
|
|
|
|
|
|
def get_logger(name: str):
|
|
log_level = getenv('LOG_LEVEL', 'INFO').upper()
|
|
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(log_level)
|
|
stream_handler = logging.StreamHandler(sys.stdout)
|
|
stream_handler.setLevel(log_level)
|
|
|
|
stream_handler.setFormatter(
|
|
logging.Formatter(
|
|
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
)
|
|
|
|
logger.addHandler(stream_handler)
|
|
|
|
return logger
|