-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathconfig.py
More file actions
32 lines (23 loc) · 792 Bytes
/
config.py
File metadata and controls
32 lines (23 loc) · 792 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import os
import yaml
class Config(object):
def __init__(self, config_path):
assert os.path.exists(config_path), "ERROR: Config File doesn't exist"
with open(config_path, 'r') as f:
self._yaml = f.read()
self._dict = yaml.load(self._yaml)
def __getattr__(self, name):
if self._dict.get(name) is not None:
return self._dict[name]
if DEFAULT_CONFIG.get(name) is not None:
return DEFAULT_CONFIG[name]
return None
def print(self):
print('Model configurations:')
print('---------------------------------')
print(self._yaml)
print('')
print('---------------------------------')
print('')
DEFAULT_CONFIG = {
}