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
1 Nov 07, 2021
2 Jan 08, 2022
534 Dec 21, 2022
5.5k Jan 04, 2023
4 Dec 31, 2022
18 Dec 02, 2022
920 Jan 04, 2023
942 Jan 05, 2023
2.8k Jan 06, 2023
9 Jan 04, 2023
2.3k Dec 30, 2022
164 Dec 27, 2022
2 Dec 10, 2021
4.6k Jan 03, 2023
1 Dec 31, 2021
127 Dec 16, 2022