提交 c1fef51d authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix warning in ReplaceValidate.replace_all_validate_remove

上级 510a9618
import inspect import inspect
import sys import sys
import time import time
import warnings
from collections import OrderedDict from collections import OrderedDict
from functools import partial from functools import partial
from io import StringIO from io import StringIO
...@@ -628,7 +629,7 @@ class ReplaceValidate(History, Validator): ...@@ -628,7 +629,7 @@ class ReplaceValidate(History, Validator):
if rm in fgraph.apply_nodes or rm in fgraph.variables: if rm in fgraph.apply_nodes or rm in fgraph.variables:
fgraph.revert(chk) fgraph.revert(chk)
if warn: if warn:
warn( warnings.warn(
"An optimization wanted to replace a Variable" "An optimization wanted to replace a Variable"
" in the graph, but the replacement for it doesn't" " in the graph, but the replacement for it doesn't"
" remove it. We disabled the optimization." " remove it. We disabled the optimization."
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论