提交 0b528507 authored 作者: Frederic's avatar Frederic

some pep8

上级 8e80196b
...@@ -514,7 +514,8 @@ class MergeFeature(object): ...@@ -514,7 +514,8 @@ class MergeFeature(object):
continue continue
inputs_match = all(node_in is cand_in inputs_match = all(node_in is cand_in
for node_in, cand_in in zip(node.inputs, candidate.inputs)) for node_in, cand_in in zip(node.inputs,
candidate.inputs))
if inputs_match and node.op == candidate.op: if inputs_match and node.op == candidate.op:
if (node, candidate) in self.blacklist: if (node, candidate) in self.blacklist:
# They were already tried, and there was an error # They were already tried, and there was an error
...@@ -811,8 +812,8 @@ class LocalOptGroup(LocalOptimizer): ...@@ -811,8 +812,8 @@ class LocalOptGroup(LocalOptimizer):
def __str__(self): def __str__(self):
return getattr(self, '__name__', return getattr(self, '__name__',
('<theano.gof.opt.LocalOptGroup instance>' ('<theano.gof.opt.LocalOptGroup instance>' +
+ str([str(o) for o in self.opts]))) str([str(o) for o in self.opts])))
def transform(self, node): def transform(self, node):
for opt in self.opts: for opt in self.opts:
...@@ -990,9 +991,9 @@ class PatternSub(LocalOptimizer): ...@@ -990,9 +991,9 @@ class PatternSub(LocalOptimizer):
else: else:
raise TypeError("The pattern to search for must start with " raise TypeError("The pattern to search for must start with "
"a specific Op instance.") "a specific Op instance.")
self.__doc__ = (self.__class__.__doc__ self.__doc__ = (self.__class__.__doc__ +
+ "\n\nThis instance does: " "\n\nThis instance does: " +
+ str(self) + "\n") str(self) + "\n")
self.allow_multiple_clients = allow_multiple_clients self.allow_multiple_clients = allow_multiple_clients
self.skip_identities_fn = skip_identities_fn self.skip_identities_fn = skip_identities_fn
if name: if name:
...@@ -1286,7 +1287,8 @@ class NavigatorOptimizer(Optimizer): ...@@ -1286,7 +1287,8 @@ class NavigatorOptimizer(Optimizer):
except Exception, e: except Exception, e:
if self.failure_callback is not None: if self.failure_callback is not None:
self.failure_callback(e, self, self.failure_callback(e, self,
[(x, None) for x in node.outputs], lopt) [(x, None) for x in node.outputs],
lopt)
return False return False
else: else:
raise raise
...@@ -1623,7 +1625,8 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1623,7 +1625,8 @@ class EquilibriumOptimizer(NavigatorOptimizer):
+ "%f with the theano flag 'optdb.max_use_ratio'." % + "%f with the theano flag 'optdb.max_use_ratio'." %
config.optdb.max_use_ratio) config.optdb.max_use_ratio)
return (self, loop_timing, loop_process_count, (start_nb_nodes, end_nb_nodes, max_nb_nodes), return (self, loop_timing, loop_process_count,
(start_nb_nodes, end_nb_nodes, max_nb_nodes),
global_opt_timing, nb_nodes, time_opts, io_toposort_timing) global_opt_timing, nb_nodes, time_opts, io_toposort_timing)
def print_summary(self, stream=sys.stdout, level=0, depth=-1): def print_summary(self, stream=sys.stdout, level=0, depth=-1):
...@@ -1637,7 +1640,8 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1637,7 +1640,8 @@ class EquilibriumOptimizer(NavigatorOptimizer):
@staticmethod @staticmethod
def print_profile(stream, prof, level=0): def print_profile(stream, prof, level=0):
(opt, loop_timing, loop_process_count, (start_nb_nodes, end_nb_nodes, max_nb_nodes), (opt, loop_timing, loop_process_count,
(start_nb_nodes, end_nb_nodes, max_nb_nodes),
global_opt_timing, nb_nodes, time_opts, io_toposort_timing) = prof global_opt_timing, nb_nodes, time_opts, io_toposort_timing) = prof
blanc = (' ' * level) blanc = (' ' * level)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论