""" Load and parse configuration file
"""
import yaml
import os

def load_config(config_file=None):
    """Loads a configuration file.

    By default it assumes ~/.ngi_config/statusdb.yaml
    """
    try:
        yaml_config = os.path.join(os.environ.get('HOME'), '.ngi_config', 'statusdb.yaml')
        if not os.path.exists(yaml_config):
            yaml_config = os.path.join(os.environ.get("STATUS_DB_CONFIG"))
        config=config_file or yaml_config
        with open(config) as f:
            conf = yaml.load(f)
            return conf
    except IOError:
        raise IOError(("There was a problem loading the configuration file. "
                "Please make sure that {} can be opened".format(config)))

""" Load and parse configuration file
"""
import yaml
import os

def load_config(config_file=None):
    """Loads a configuration file.

    By default it assumes ~/.ngi_config/statusdb.yaml
    """
    try:
        yaml_config = os.path.join(os.environ.get('HOME'), '.ngi_config', 'statusdb.yaml')
        if not os.path.exists(yaml_config):
            yaml_config = os.path.join(os.environ.get("STATUS_DB_CONFIG"))
        config=config_file or yaml_config
        with open(config) as f:
            conf = yaml.load(f, Loader=yaml.SafeLoader)
            return conf
    except IOError:
        raise IOError(("There was a problem loading the configuration file. "
                "Please make sure that {} can be opened".format(config)))

Update yaml.load