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