Logging: Switch to loguru

Loguru is a flexible logger that allows for easier hooking and imports
into Rich with no problems. Also makes progress bars stick to the
bottom of the terminal window.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-03-07 23:20:17 -05:00
committed by Brian Dashore
parent fe0ff240e7
commit 228c227c1e
14 changed files with 110 additions and 119 deletions

View File

@@ -1,17 +1,14 @@
"""Common functions for sampling parameters"""
import pathlib
from typing import Dict, List, Optional, Union
from pydantic import AliasChoices, BaseModel, Field
import yaml
from loguru import logger
from pydantic import AliasChoices, BaseModel, Field
from typing import Dict, List, Optional, Union
from common.logger import init_logger
from common.utils import unwrap, prune_dict
logger = init_logger(__name__)
# Common class for sampler params
class BaseSamplerRequest(BaseModel):
"""Common class for sampler params that are used in APIs"""