-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdowel_wrapper.py
33 lines (27 loc) · 894 Bytes
/
dowel_wrapper.py
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
33
import sys
assert 'dowel' not in sys.modules, 'dowel must be imported after dowel_wrapper.'
import dowel
dowel_eval = dowel
del sys.modules['dowel']
import dowel
dowel_plot = dowel
del sys.modules['dowel']
import dowel
dowel_tcp = dowel
del sys.modules['dowel']
import dowel
all_dowels = [dowel, dowel_eval, dowel_plot, dowel_tcp]
assert len(set(id(d) for d in all_dowels)) == len(all_dowels)
import aim_wrapper
def get_dowel(phase=None):
if (phase or aim_wrapper.get_context().get('phase')).lower() == 'plot':
return dowel_plot
if (phase or aim_wrapper.get_context().get('phase')).lower() == 'eval':
return dowel_eval
if (phase or aim_wrapper.get_context().get('phase')).lower() == 'tcp':
return dowel_tcp
return dowel
def get_logger(phase=None):
return get_dowel(phase).logger
def get_tabular(phase=None):
return get_dowel(phase).tabular