提交 15f6a12c authored 作者: Jakub Sygnowski's avatar Jakub Sygnowski

getattr in vm to allow for a new attribute

上级 c4d1145b
......@@ -778,9 +778,6 @@ class Function(object):
output_subset = kwargs.pop('output_subset', None)
if output_subset is not None and self.output_keys is not None:
if not all(key in self.output_keys for key in output_subset):
raise KeyError('Output_subset should be a list of keys '
'from output_keys')
output_subset =\
[self.output_keys.index(key) for key in output_subset]
......@@ -980,9 +977,6 @@ class Function(object):
if output_subset is None:
return outputs
else:
if not all(isinstance(index, int) for index in output_subset):
raise TypeError('Output_subset should be '
'a list of indices of output variables')
return [outputs[i] for i in output_subset]
value = property(
......
......@@ -821,14 +821,14 @@ class VM_Linker(link.LocalLinker):
if (self.callback is not None or
(config.profile and config.profile_memory) or
self.allow_partial_eval):
getattr(self, 'allow_partial_eval', False)):
if self.use_cloop and self.callback is not None:
logger.warn('CVM does not support callback, using Stack VM.')
if self.use_cloop and config.profile_memory:
warnings.warn(
'CVM does not support memory profile, using Stack VM.')
if self.use_cloop and self.allow_partial_eval:
if self.use_cloop and getattr(self, 'allow_partial_eval', False):
warnings.warn(
'CVM does not support partial evaluation yet, '
'using Stack VM.')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论