From d49a5c1be21e2741dfa48f1e236e8962758a0204 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 8 Sep 2020 16:45:47 +0800 Subject: [PATCH] fix(mge/imperative): remove codes newly introduced in numpy 1.19 GitOrigin-RevId: 874350cbc78d449eca4179ffcd49807afeafc3f6 --- imperative/python/megengine/jit/tracing.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/imperative/python/megengine/jit/tracing.py b/imperative/python/megengine/jit/tracing.py index 2c116f29..baa40c34 100644 --- a/imperative/python/megengine/jit/tracing.py +++ b/imperative/python/megengine/jit/tracing.py @@ -155,9 +155,7 @@ class trace: "last time, got an internal tensor this time" ) if x._handle != info.bound_data._handle: - if not np.array_equal( - x.numpy(), info.bound_data.numpy(), equal_nan=True - ): + if not np.array_equal(x.numpy(), info.bound_data.numpy()): raise TraceMismatchError( "const capture violated: got " "a different tensor this time" -- GitLab