A Pydantic-ish way to manage your project's YAML configurations.
Documentation: https://dribia.github.io/driconfig
Source Code: https://github.com/dribia/driconfig
The usage of YAML files to store configurations and parameters is widely accepted in the Python community, especially in Data Science environments.
DriConfig provides a clean interface between your Python code and these YAML configuration files.
It is heavily based on Pydantic's Settings Management, preserving its core functionalities and advantages.
Key features
- Subclassing the
DriConfigclass we create an interface to any YAML configuration file. - Our project's configurations are then attributes of this class.
- They are automatically filled with the values in the YAML configuration file.
- We can define complex configuration structures using Pydantic models.
- We preserve Pydantic's type casting and validation!
Example
Let's say we have a YAML configuration file config.yaml with the following data:
# config.yaml
model_parameters:
eta: 0.2
gamma: 2
lambda: 1
date_interval:
start: 2021-01-01
end: 2021-12-31
Then we can parse with driconfig as follows:
from datetime import date
from typing import Dict
from driconfig import DriConfig
from pydantic import BaseModel
class DateInterval(BaseModel):
"""Model for the `date_interval` configuration."""
start: date
end: date
class AppConfig(DriConfig):
"""Interface for the config/config.yaml file."""
class Config:
"""Configure the YAML file location."""
config_folder = "."
config_file_name = "config.yaml"
model_parameters: Dict[str, float]
date_interval: DateInterval
config = AppConfig()
print(config.json(indent=4))
"""
{
"model_parameters": {
"eta": 0.2,
"gamma": 2.0,
"lambda": 1.0
},
"date_interval": {
"start": "2021-01-01",
"end": "2021-12-31"
}
}
"""
6 Jun 29, 2022
0 Nov 21, 2022
7 Dec 27, 2022
8 Dec 14, 2022
1 Dec 17, 2021
3 Sep 18, 2022
11 Sep 15, 2022
3 Feb 19, 2022
24 Feb 16, 2022
2 Jan 08, 2022
2 Oct 09, 2021
1 Nov 09, 2021
955 Jan 05, 2023
4.6k Jan 03, 2023
9 Jan 04, 2023
5 Dec 23, 2022
532 Dec 27, 2022
534 Dec 21, 2022
18 Dec 02, 2022
75 Sep 09, 2022
920 Jan 04, 2023
127 Dec 16, 2022
38 Nov 01, 2022
2.7k Jan 03, 2023
2 Jan 08, 2022