From 0a5d2a14928c992a5e43a9a82ad04af227bb8fe8 Mon Sep 17 00:00:00 2001
From: boringbyte <kottapallisivaram@gmail.com>
Date: Sat, 7 Dec 2024 08:47:37 -0500
Subject: [PATCH] fix type annotations for 'optuna/storages/journal/_redis.py'

---
 optuna/storages/journal/_redis.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/optuna/storages/journal/_redis.py b/optuna/storages/journal/_redis.py
index d45486c420..811cdc1eb4 100644
--- a/optuna/storages/journal/_redis.py
+++ b/optuna/storages/journal/_redis.py
@@ -1,9 +1,8 @@
+from __future__ import annotations
+
 import json
 import time
 from typing import Any
-from typing import Dict
-from typing import List
-from typing import Optional
 
 from optuna._deprecated import deprecated_class
 from optuna._experimental import experimental_class
@@ -44,16 +43,16 @@ def __init__(self, url: str, use_cluster: bool = False, prefix: str = "") -> Non
         self._use_cluster = use_cluster
         self._prefix = prefix
 
-    def __getstate__(self) -> Dict[Any, Any]:
+    def __getstate__(self) -> dict[Any, Any]:
         state = self.__dict__.copy()
         del state["_redis"]
         return state
 
-    def __setstate__(self, state: Dict[Any, Any]) -> None:
+    def __setstate__(self, state: dict[Any, Any]) -> None:
         self.__dict__.update(state)
         self._redis = redis.Redis.from_url(self._url)
 
-    def read_logs(self, log_number_from: int) -> List[Dict[str, Any]]:
+    def read_logs(self, log_number_from: int) -> list[dict[str, Any]]:
         max_log_number_bytes = self._redis.get(f"{self._prefix}:log_number")
         if max_log_number_bytes is None:
             return []
@@ -75,7 +74,7 @@ def read_logs(self, log_number_from: int) -> List[Dict[str, Any]]:
                     raise err
         return logs
 
-    def append_logs(self, logs: List[Dict[str, Any]]) -> None:
+    def append_logs(self, logs: list[dict[str, Any]]) -> None:
         self._redis.setnx(f"{self._prefix}:log_number", -1)
         for log in logs:
             if not self._use_cluster:
@@ -93,7 +92,7 @@ def append_logs(self, logs: List[Dict[str, Any]]) -> None:
     def save_snapshot(self, snapshot: bytes) -> None:
         self._redis.set(f"{self._prefix}:snapshot", snapshot)
 
-    def load_snapshot(self) -> Optional[bytes]:
+    def load_snapshot(self) -> bytes | None:
         snapshot_bytes = self._redis.get(f"{self._prefix}:snapshot")
         return snapshot_bytes