提交 daedf052 authored 作者: Frederic Bastien's avatar Frederic Bastien

Allow change_flags to be used for flags with sections.

上级 aaf96699
......@@ -98,15 +98,17 @@ class change_flags(object):
Useful during tests.
"""
def __init__(self, **kwargs):
def __init__(self, *args, **kwargs):
confs = dict()
for k in kwargs:
args = dict(*args)
args.update(kwargs)
for k in args:
l = [v for v in theano.configparser._config_var_list
if v.fullname == k]
assert len(l) == 1
confs[k] = l[0]
self.confs = confs
self.new_vals = kwargs
self.new_vals = args
def __call__(self, f):
@wraps(f)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论