Ignore an incorrect mypy error.

gijskoning · Oct 7, 2020 · ab7436a · ab7436a
1 parent c88be87
commit ab7436a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jax/linear_util.py b/jax/linear_util.py
@@ -247,7 +247,7 @@ def _most_recent_entry():
       thread_local.most_recent_entry = None
       return result
 
-  memoized_fun.most_recent_entry = _most_recent_entry
+  memoized_fun.most_recent_entry = _most_recent_entry  # type: ignore
   memoized_fun.cache_clear = fun_caches.clear  # type: ignore
 
   return memoized_fun

diff --git a/tests/jax_jit_test.py b/tests/jax_jit_test.py
@@ -82,7 +82,7 @@ def test_convert_scalars(self):
     self.assertEqual(jnp.asarray(1 + 1j).dtype, res.dtype)
 
   def test_signature_support(self):
-    if version < (0, 1, 54):
+    if version < (0, 1, 56):
       return
 
     def f(a, b, c):