common.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

from collections import defaultdict

def load_params(fname):
    params = defaultdict(int)
    with open(fname, 'r') as f:
        data = f.read()
    try:
        parsed = eval(data)
        # Convert to defaultdict
        for k in parsed:
            params[k] = str(parsed[k])
    except Exception as e:
        raise IOError("Invalid param file: %s\n%s" % (fname, e))

    return params