提交 152aa23e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix typo/logic problems in the treatment of local optimizations that apply to all nodes.

上级 1efa6ac3
...@@ -780,8 +780,6 @@ class LocalOptimizer(object): ...@@ -780,8 +780,6 @@ class LocalOptimizer(object):
class FromFunctionLocalOptimizer(LocalOptimizer): class FromFunctionLocalOptimizer(LocalOptimizer):
"""WRITEME""" """WRITEME"""
def __init__(self, fn, tracks=None): def __init__(self, fn, tracks=None):
if tracks is None:
tracks = []
self.transform = fn self.transform = fn
self._tracks = tracks self._tracks = tracks
...@@ -1510,7 +1508,7 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1510,7 +1508,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
for opt in optimizers: for opt in optimizers:
if isinstance(opt, LocalOptimizer): if isinstance(opt, LocalOptimizer):
if opt.tracks is None: if opt.tracks() is None:
self.local_optimizers_all.append(opt) self.local_optimizers_all.append(opt)
else: else:
for c in opt.tracks(): for c in opt.tracks():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论