samgis / samgis /__init__.py
aletrn's picture
[refactor] first working stage for huggingface docker container with fastapi and loguru
7c6fa68
raw
history blame
1.28 kB
"""Get machine learning predictions from geodata raster images"""
# not used here but contextily_tile is imported in samgis.io.tms2geotiff
from contextily import tile as contextily_tile
from pathlib import Path
from samgis.utilities.constants import SERVICE_NAME
PROJECT_ROOT_FOLDER = Path(globals().get("__file__", "./_")).absolute().parent.parent
MODEL_FOLDER = Path(PROJECT_ROOT_FOLDER / "machine_learning_models")
try:
from aws_lambda_powertools import Logger
app_logger = Logger(service=SERVICE_NAME)
except ModuleNotFoundError:
import loguru
def setup_logging(debug: bool = False, formatter: str = "{time} - {level} - ({extra[request_id]}) {message} "
) -> loguru.logger:
"""
Create a logging instance with log string formatter.
Args:
debug: logging debug argument
formatter: log string formatter
Returns:
Logger
"""
import sys
logger = loguru.logger
logger.remove()
level_logger = "DEBUG" if debug else "INFO"
logger.add(sys.stdout, format=formatter, level=level_logger)
logger.info(f"type_logger:{type(logger)}, logger:{logger}.")
return logger
app_logger = setup_logging(debug=True)