From f46d41f2c298de14e352dde472e9687653254798 Mon Sep 17 00:00:00 2001
From: Ayaka <ayx@google.com>
Date: Tue, 11 Jun 2024 16:14:16 +0400
Subject: [PATCH] Try fix test error by reverting adding type annotations

---
 jax/_src/compilation_cache_interface.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/jax/_src/compilation_cache_interface.py b/jax/_src/compilation_cache_interface.py
index eeb5b01c8f3e..95d557c5531e 100644
--- a/jax/_src/compilation_cache_interface.py
+++ b/jax/_src/compilation_cache_interface.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from __future__ import annotations
-
 from abc import abstractmethod
 
 from jax._src import path as pathlib
@@ -24,9 +22,9 @@ class CacheInterface(util.StrictABC):
   _path: pathlib.Path
 
   @abstractmethod
-  def get(self, key: str) -> bytes | None:
+  def get(self, key: str):
     pass
 
   @abstractmethod
-  def put(self, key: str, value: bytes) -> None:
+  def put(self, key: str, value: bytes):
     pass