Initialize wandb and import to setup notification system

log_wandb.py

import wandb


def init_wandb(project, name=None, notes=None, config=None):
    wandb.login()
    run = wandb.init(
        project=project,
        name=name,
        notes=notes,
        config=config,
    )
    return run