diff --git a/jax/_src/lru_cache.py b/jax/_src/lru_cache.py
index c961e6cfc0a6..692e51481fb5 100644
--- a/jax/_src/lru_cache.py
+++ b/jax/_src/lru_cache.py
@@ -23,7 +23,7 @@
 
 
 try:
-  import filelock
+  import filelock  # pytype: disable=import-error
 except ImportError:
   filelock = None