From 079b4a6b50c21f88e9b70ec842d15a833d3af913 Mon Sep 17 00:00:00 2001
From: Ayaka <ayx@google.com>
Date: Tue, 11 Jun 2024 16:17:34 +0400
Subject: [PATCH] Fix test

---
 jax/_src/lru_cache.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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