提交 a89390c1 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Correctly restore values_eq_approx

上级 58328cd7
......@@ -4637,7 +4637,7 @@ class T_local_switch_sink(unittest.TestCase):
def wrapped_f(*args, **kwargs):
# This is a bit ugly since it changes the global value of
# TensorType.values_eq_approx.
old_values_eq_approx = TensorType.values_eq_approx
old_values_eq_approx = staticmethod(TensorType.values_eq_approx)
TensorType.values_eq_approx = staticmethod(values_eq_approx_remove_nan)
try:
out = f(*args, **kwargs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论