YamlAttributes

YamlAttributes is a utility class to load yaml files and assign the entries to associated class attributes. This can be used to create a type safe and dynamic configuration class.

💾 Installation

YamlAttributes can be installed by running pip install yamlattributes

🧑‍💻 Usage

The abstract YamlAttributes class provides the init method which loads a YAML file and assigns its values to the corresponding class attributes. A class which inherits from YamlAttributes simply has to list the desired configuration as class attributes and two additional special attributes yaml_file_path and yaml_section. After calling the init method the all class attributes wil be assigned the values of the corresponding fields of the YAML file.

🧐 API

Special Attributes

Attribute	Default	Description
`yaml_file_path`	`'./'`	Sets the path to the YAML file which should be loaded.
`yaml_section`	`'config'`	Sets YAML section which contains the desired config fields.

Optional Attributes

By default the YAML file has to provide values for all attributes of the config file. By using the Optional type hint optional attributes can be added to a config class. For these optional attributes the YAML file is not required to provide values.

Methods

`init`

Args	values	Description
`mode`	`'sync'`, `'soft_config'`	Selects the the mode of how the YAML values should update the config class. The `sync` mode keeps the YAML file and the config class in sync. Each attribute in the config class (except of the two special ones and optiona attributes) have to be in the YAML file and vis versa. The `soft_config` mode allows the YAML file to have additional fields which the config class does not have.
`yaml_file_path`	`str`	Instead of providing the path to the YAML file through the special attribute it can also be passed in the init method. The path passed to the init method overrides the special attribute
`yaml_section`	`str`	Instead of providing the YAML section through the special attribute it can also be passed in the init method. The section passed to the init method overrides the special attribute

`to_dict`

The config object can be converted to a dictionary by calling the to_dict method

🚀 Example

First create your desired config class and set the destination of the YAML file to be loaded through the special yaml_file_path attribute.

# ./config.py

class MyConfig(YamlAttributes):
    yaml_file_path = './configs/my-config.yaml'
    batch_size: int
    num_classes: int
    optimizer: str
    device: Optional[str]

Make ure the YAML file exists and has all entries the config class MyConfig needs.

# ./configs/my-config.yaml

config:
    batch_size: 32
    num_classes: 10
    optimizer: 'adam'

Finally you can initialize your config class and enjoy accessing config values with autocomplete.

# ./main.py

from config import MyConfig

MyConfig.init()

# Access config values with autocomplete
MyConfig.batch_size

🛠️ Contributing

Interested in contributing? Great, thanks for your interest!

To fix a bug or add a feature, follow these steps:

Fork the repo
Create a new branch (git checkout -b your-branch)
Add your changes and new tests if necessary
Make sure all tests pass
Commit your changes (git commit -am 'feat: fantastic feature')
Push the branch (git push origin your-branch)
Create a Pull Request

Developing

The required packages to start development can be installed with

pipenv install --dev

The tests can be run with

pipenv run pytest

To build the package run

python3 -m build

You can then install it locally with

pip install .

gcascio/yaml-attributes