bb6b2d059ad7b8a231d9f8015b41e60f033eaf76,studio/model.py,,get_config,#Any#,595

Before Change

def get_config(config_file=None):

    if config_file:
        with open(config_file) as f:
            config = yaml.load(f.read())
    else:
        def_config_file = os.path.join(
            os.path.dirname(os.path.realpath(__file__)),
            "default_config.yaml")

After Change

def get_config(config_file=None):

    config_paths = []
    if config_file:
        config_paths.append(os.expanduser(config_file))
    
    config_paths.append(os.path.expanduser("~/.tfstudio/config.yaml"))
    config_paths.append(
            os.path.join(
            os.path.dirname(os.path.realpath(__file__)),
            "default_config.yaml"))

    for path in config_paths:
        if not os.path.exists(path):
            continue
        
        with(open(path)) as f:
            return yaml.load(f.read())

    raise ValueError("None of the config paths {} exits!"
            .format(config_paths))

Instances