Spaces:
No application file
No application file
File size: 704 Bytes
480e694 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import logging
from pathlib import Path
from typing import Optional
def setup_logging(name: str, level: str = "INFO") -> logging.Logger:
logger = logging.getLogger(name)
logger.setLevel(level)
if not logger.handlers:
handler = logging.StreamHandler()
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
logger.addHandler(handler)
return logger
def ensure_model_path(path: Optional[str] = None) -> Path:
if path is None:
path = "../models"
model_path = Path(path)
model_path.mkdir(parents=True, exist_ok=True)
return model_path
|