ArgBind demo

demo.py

import argbind
from typing import List

@argbind.bind()
def train(epochs : int = 50):
    """Run training.

    Parameters
    ----------
    epochs : int, optional
        Number of epochs, by default 50
    """
    print('train', epochs)

@argbind.bind()
def evaluate(save_folder : str = 'folder'):
    """Run evaluation.

    Parameters
    ----------
    save_folder : str, optional
        Where to save, by default '.'
    """
    print('evaluate', save_folder)

@argbind.bind()
def run(stages : List[str] = ['train', 'evaluate']):
    """Run stages.

    Parameters
    ----------
    stages : List[str], optional
        Which stages to run, by default 'train'
    """
    for stage in stages:
        stage_fn = globals()[stage]
        stage_fn() # will do train

if __name__ == "__main__":
    args = argbind.parse_args()
    with argbind.scope(args):
        run()

❯ python demo.py --train.epochs=10 --evaluate.save_folder=different_folder --args.save demo.yml
train 10
evaluate different_folder
❯ python demo.py --args.load demo.yml
train 10
evaluate different_folder

Contents of demo.yml:

evaluate.save_folder: different_folder

run.stages:
- train
- evaluate

train.epochs: 10