streamlit-nightly 1.31.2.dev20240213__py2.py3-none-any.whl → 1.31.2.dev20240214__py2.py3-none-any.whl

streamlit/runtime/app_session.py

@@ -11,11 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from __future__ import annotations
+
 import asyncio
 import sys
 import uuid
 from enum import Enum
-from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Callable, Final
 
 import streamlit.elements.exception as exception_utils
 from streamlit import config, runtime, source_util
@@ -44,10 +47,11 @@ from streamlit.runtime.uploaded_file_manager import UploadedFileManager
 from streamlit.version import STREAMLIT_VERSION_STRING
 from streamlit.watcher import LocalSourcesWatcher
 
-LOGGER = get_logger(__name__)
 if TYPE_CHECKING:
     from streamlit.runtime.state import SessionState
 
+_LOGGER: Final = get_logger(__name__)
+
 
 class AppSessionState(Enum):
     APP_NOT_RUNNING = "APP_NOT_RUNNING"
@@ -77,10 +81,10 @@ class AppSession:
         script_data: ScriptData,
         uploaded_file_manager: UploadedFileManager,
         script_cache: ScriptCache,
-        message_enqueued_callback: Optional[Callable[[], None]],
+        message_enqueued_callback: Callable[[], None] | None,
         local_sources_watcher: LocalSourcesWatcher,
-        user_info: Dict[str, Optional[str]],
-        session_id_override: Optional[str] = None,
+        user_info: dict[str, str | None],
+        session_id_override: str | None = None,
     ) -> None:
         """Initialize the AppSession.
 
@@ -139,17 +143,15 @@ class AppSession:
         # due to the source code changing we need to pass in the previous client state.
         self._client_state = ClientState()
 
-        self._local_sources_watcher: Optional[
-            LocalSourcesWatcher
-        ] = local_sources_watcher
-        self._stop_config_listener: Optional[Callable[[], bool]] = None
-        self._stop_pages_listener: Optional[Callable[[], bool]] = None
+        self._local_sources_watcher: LocalSourcesWatcher | None = local_sources_watcher
+        self._stop_config_listener: Callable[[], bool] | None = None
+        self._stop_pages_listener: Callable[[], None] | None = None
 
         self.register_file_watchers()
 
         self._run_on_save = config.get_option("server.runOnSave")
 
-        self._scriptrunner: Optional[ScriptRunner] = None
+        self._scriptrunner: ScriptRunner | None = None
 
         # This needs to be lazily imported to avoid a dependency cycle.
         from streamlit.runtime.state import SessionState
@@ -157,9 +159,9 @@ class AppSession:
         self._session_state = SessionState()
         self._user_info = user_info
 
-        self._debug_last_backmsg_id: Optional[str] = None
+        self._debug_last_backmsg_id: str | None = None
 
-        LOGGER.debug("AppSession initialized (id=%s)", self.id)
+        _LOGGER.debug("AppSession initialized (id=%s)", self.id)
 
     def __del__(self) -> None:
         """Ensure that we call shutdown() when an AppSession is garbage collected."""
@@ -210,7 +212,7 @@ class AppSession:
         self._stop_config_listener = None
         self._stop_pages_listener = None
 
-    def flush_browser_queue(self) -> List[ForwardMsg]:
+    def flush_browser_queue(self) -> list[ForwardMsg]:
         """Clear the forward message queue and return the messages it contained.
 
         The Server calls this periodically to deliver new messages
@@ -232,7 +234,7 @@ class AppSession:
 
         """
         if self._state != AppSessionState.SHUTDOWN_REQUESTED:
-            LOGGER.debug("Shutting down (id=%s)", self.id)
+            _LOGGER.debug("Shutting down (id=%s)", self.id)
             # Clear any unused session files in upload file manager and media
             # file manager
             self._uploaded_file_mgr.remove_session_files(self.id)
@@ -298,10 +300,10 @@ class AppSession:
             elif msg_type == "file_urls_request":
                 self._handle_file_urls_request(msg.file_urls_request)
             else:
-                LOGGER.warning('No handler for "%s"', msg_type)
+                _LOGGER.warning('No handler for "%s"', msg_type)
 
         except Exception as ex:
-            LOGGER.error(ex)
+            _LOGGER.error(ex)
             self.handle_backmsg_exception(ex)
 
     def handle_backmsg_exception(self, e: BaseException) -> None:
@@ -333,7 +335,7 @@ class AppSession:
             lambda: self._enqueue_forward_msg(self._create_exception_message(e))
         )
 
-    def request_rerun(self, client_state: Optional[ClientState]) -> None:
+    def request_rerun(self, client_state: ClientState | None) -> None:
         """Signal that we're interested in running the script.
 
         If the script is not already running, it will be started immediately.
@@ -347,7 +349,7 @@ class AppSession:
 
         """
         if self._state == AppSessionState.SHUTDOWN_REQUESTED:
-            LOGGER.warning("Discarding rerun request after shutdown")
+            _LOGGER.warning("Discarding rerun request after shutdown")
             return
 
         if client_state:
@@ -403,7 +405,7 @@ class AppSession:
         self._scriptrunner.start()
 
     @property
-    def session_state(self) -> "SessionState":
+    def session_state(self) -> SessionState:
         return self._session_state
 
     def _should_rerun_on_file_change(self, filepath: str) -> bool:
@@ -421,7 +423,7 @@ class AppSession:
 
         return True
 
-    def _on_source_file_changed(self, filepath: Optional[str] = None) -> None:
+    def _on_source_file_changed(self, filepath: str | None = None) -> None:
         """One of our source files changed. Clear the cache and schedule a rerun if appropriate."""
         self._script_cache.clear()
 
@@ -456,12 +458,12 @@ class AppSession:
 
     def _on_scriptrunner_event(
         self,
-        sender: Optional[ScriptRunner],
+        sender: ScriptRunner | None,
         event: ScriptRunnerEvent,
-        forward_msg: Optional[ForwardMsg] = None,
-        exception: Optional[BaseException] = None,
-        client_state: Optional[ClientState] = None,
-        page_script_hash: Optional[str] = None,
+        forward_msg: ForwardMsg | None = None,
+        exception: BaseException | None = None,
+        client_state: ClientState | None = None,
+        page_script_hash: str | None = None,
     ) -> None:
         """Called when our ScriptRunner emits an event.
 
@@ -477,12 +479,12 @@ class AppSession:
 
     def _handle_scriptrunner_event_on_event_loop(
         self,
-        sender: Optional[ScriptRunner],
+        sender: ScriptRunner | None,
         event: ScriptRunnerEvent,
-        forward_msg: Optional[ForwardMsg] = None,
-        exception: Optional[BaseException] = None,
-        client_state: Optional[ClientState] = None,
-        page_script_hash: Optional[str] = None,
+        forward_msg: ForwardMsg | None = None,
+        exception: BaseException | None = None,
+        client_state: ClientState | None = None,
+        page_script_hash: str | None = None,
     ) -> None:
         """Handle a ScriptRunner event.
 
@@ -525,7 +527,7 @@ class AppSession:
             # rerun request, for example) while another ScriptRunner is still
             # shutting down. The shutting-down ScriptRunner may still
             # emit events.
-            LOGGER.debug("Ignoring event from non-current ScriptRunner: %s", event)
+            _LOGGER.debug("Ignoring event from non-current ScriptRunner: %s", event)
             return
 
         prev_state = self._state
@@ -663,7 +665,7 @@ class AppSession:
         return msg
 
     def _create_script_finished_message(
-        self, status: "ForwardMsg.ScriptFinishedStatus.ValueType"
+        self, status: ForwardMsg.ScriptFinishedStatus.ValueType
     ) -> ForwardMsg:
         """Create and return a script_finished ForwardMsg."""
         msg = ForwardMsg()
@@ -712,10 +714,10 @@ class AppSession:
         except Exception as ex:
             # Users may never even install Git in the first place, so this
             # error requires no action. It can be useful for debugging.
-            LOGGER.debug("Obtaining Git information produced an error", exc_info=ex)
+            _LOGGER.debug("Obtaining Git information produced an error", exc_info=ex)
 
     def _handle_rerun_script_request(
-        self, client_state: Optional[ClientState] = None
+        self, client_state: ClientState | None = None
     ) -> None:
         """Tell the ScriptRunner to re-run its script.
 
@@ -794,10 +796,10 @@ class AppSession:
 # This field will be available at runtime as of protobuf 3.20.1, but
 # we are using an older version.
 # For details, see: https://github.com/protocolbuffers/protobuf/issues/8175
-def _get_toolbar_mode() -> "Config.ToolbarMode.ValueType":
+def _get_toolbar_mode() -> Config.ToolbarMode.ValueType:
     config_key = "client.toolbarMode"
     config_value = config.get_option(config_key)
-    enum_value: Optional["Config.ToolbarMode.ValueType"] = getattr(
+    enum_value: Config.ToolbarMode.ValueType | None = getattr(
         Config.ToolbarMode, config_value.upper()
     )
     if enum_value is None:
@@ -844,7 +846,7 @@ def _populate_theme_msg(msg: CustomThemeConfig) -> None:
     base = theme_opts["base"]
     if base is not None:
         if base not in base_map:
-            LOGGER.warning(
+            _LOGGER.warning(
                 f'"{base}" is an invalid value for theme.base.'
                 f" Allowed values include {list(base_map.keys())}."
                 ' Setting theme.base to "light".'
@@ -860,7 +862,7 @@ def _populate_theme_msg(msg: CustomThemeConfig) -> None:
     font = theme_opts["font"]
     if font is not None:
         if font not in font_map:
-            LOGGER.warning(
+            _LOGGER.warning(
                 f'"{font}" is an invalid value for theme.font.'
                 f" Allowed values include {list(font_map.keys())}."
                 ' Setting theme.font to "sans serif".'
@@ -874,9 +876,7 @@ def _populate_user_info_msg(msg: UserInfo) -> None:
     msg.installation_id_v3 = Installation.instance().installation_id_v3
 
 
-def _populate_app_pages(
-    msg: Union[NewSession, PagesChanged], main_script_path: str
-) -> None:
+def _populate_app_pages(msg: NewSession | PagesChanged, main_script_path: str) -> None:
     for page_script_hash, page_info in source_util.get_pages(main_script_path).items():
         page_proto = msg.app_pages.add()
 

streamlit/runtime/caching/__init__.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import contextlib
-from typing import Any, Iterator, Union
+from typing import Any, Iterator
 
 from google.protobuf.message import Message
 
@@ -77,9 +79,7 @@ def save_widget_metadata(metadata: WidgetMetadata[Any]) -> None:
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX.save_widget_metadata(metadata)
 
 
-def save_media_data(
-    image_data: Union[bytes, str], mimetype: str, image_id: str
-) -> None:
+def save_media_data(image_data: bytes | str, mimetype: str, image_id: str) -> None:
     CACHE_DATA_MESSAGE_REPLAY_CTX.save_image_data(image_data, mimetype, image_id)
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX.save_image_data(image_data, mimetype, image_id)
 

streamlit/runtime/caching/cache_data_api.py

@@ -20,9 +20,9 @@ import pickle
 import threading
 import types
 from datetime import timedelta
-from typing import Any, Callable, TypeVar, Union, cast, overload
+from typing import Any, Callable, Final, Literal, TypeVar, Union, cast, overload
 
-from typing_extensions import Literal, TypeAlias
+from typing_extensions import TypeAlias
 
 import streamlit as st
 from streamlit import runtime
@@ -62,7 +62,7 @@ from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner.script_run_context import get_script_run_ctx
 from streamlit.runtime.stats import CacheStat, CacheStatsProvider, group_stats
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 CACHE_DATA_MESSAGE_REPLAY_CTX = CachedMessageReplayContext(CacheType.DATA)
 

streamlit/runtime/caching/cache_errors.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import types
-from typing import Any, Optional
+from typing import Any
 
 from streamlit import type_util
 from streamlit.errors import (
@@ -50,7 +52,7 @@ class UnhashableParamError(StreamlitAPIException):
         self,
         cache_type: CacheType,
         func: types.FunctionType,
-        arg_name: Optional[str],
+        arg_name: str | None,
         arg_value: Any,
         orig_exc: BaseException,
     ):
@@ -62,7 +64,7 @@ class UnhashableParamError(StreamlitAPIException):
     def _create_message(
         cache_type: CacheType,
         func: types.FunctionType,
-        arg_name: Optional[str],
+        arg_name: str | None,
         arg_value: Any,
     ) -> str:
         arg_name_str = arg_name if arg_name is not None else "(unnamed)"

streamlit/runtime/caching/cache_type.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import enum
 
 

streamlit/runtime/caching/cache_utils.py

@@ -26,9 +26,7 @@ import types
 from abc import abstractmethod
 from collections import defaultdict
 from datetime import timedelta
-from typing import Any, Callable, overload
-
-from typing_extensions import Literal
+from typing import Any, Callable, Final, Literal, overload
 
 from streamlit import type_util
 from streamlit.elements.spinner import spinner
@@ -53,7 +51,7 @@ from streamlit.runtime.caching.cached_message_replay import (
 from streamlit.runtime.caching.hashing import HashFuncsDict, update_hash
 from streamlit.util import HASHLIB_KWARGS
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 # The timer function we use with TTLCache. This is the default timer func, but
 # is exposed here as a constant so that it can be patched in unit tests.

streamlit/runtime/caching/cached_message_replay.py

@@ -19,10 +19,17 @@ import hashlib
 import threading
 import types
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Iterator, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Final,
+    Iterator,
+    Protocol,
+    Union,
+    runtime_checkable,
+)
 
 from google.protobuf.message import Message
-from typing_extensions import Protocol, runtime_checkable
 
 import streamlit as st
 from streamlit import runtime, util
@@ -45,7 +52,7 @@ from streamlit.util import HASHLIB_KWARGS
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
 @runtime_checkable
@@ -369,7 +376,7 @@ class CachedMessageReplayContext(threading.local):
 
     def maybe_show_cached_st_function_warning(
         self,
-        dg: "DeltaGenerator",
+        dg: DeltaGenerator,
         st_func_name: str,
     ) -> None:
         """If appropriate, warn about calling st.foo inside @memo.
@@ -401,7 +408,7 @@ class CachedMessageReplayContext(threading.local):
 
     def _show_cached_st_function_warning(
         self,
-        dg: "DeltaGenerator",
+        dg: DeltaGenerator,
         st_func_name: str,
         cached_func: types.FunctionType,
     ) -> None:

streamlit/runtime/caching/storage/cache_storage_protocol.py

@@ -57,8 +57,7 @@ from __future__ import annotations
 
 from abc import abstractmethod
 from dataclasses import dataclass
-
-from typing_extensions import Literal, Protocol
+from typing import Literal, Protocol
 
 
 class CacheStorageError(Exception):

streamlit/runtime/caching/storage/local_disk_cache_storage.py

@@ -61,6 +61,7 @@ from __future__ import annotations
 import math
 import os
 import shutil
+from typing import Final
 
 from streamlit import util
 from streamlit.file_util import get_streamlit_file_path, streamlit_read, streamlit_write
@@ -76,16 +77,16 @@ from streamlit.runtime.caching.storage.in_memory_cache_storage_wrapper import (
     InMemoryCacheStorageWrapper,
 )
 
+_LOGGER: Final = get_logger(__name__)
+
 # Streamlit directory where persisted @st.cache_data objects live.
 # (This is the same directory that @st.cache persisted objects live.
 # But @st.cache_data uses a different extension, so they don't overlap.)
-_CACHE_DIR_NAME = "cache"
+_CACHE_DIR_NAME: Final = "cache"
 
 # The extension for our persisted @st.cache_data objects.
 # (`@st.cache_data` was originally called `@st.memo`)
-_CACHED_FILE_EXTENSION = "memo"
-
-_LOGGER = get_logger(__name__)
+_CACHED_FILE_EXTENSION: Final = "memo"
 
 
 class LocalDiskCacheStorageManager(CacheStorageManager):
@@ -168,7 +169,7 @@ class LocalDiskCacheStorage(CacheStorage):
                 # Clean up file so we don't leave zero byte files.
                 try:
                     os.remove(path)
-                except (FileNotFoundError, IOError, OSError):
+                except (FileNotFoundError, OSError):
                     # If we can't remove the file, it's not a big deal.
                     pass
                 raise CacheStorageError("Unable to write to cache") from e

streamlit/runtime/connection_factory.py

@@ -14,13 +14,10 @@
 
 from __future__ import annotations
 
-import importlib
 import os
 import re
 from datetime import timedelta
-from typing import Any, Dict, Type, TypeVar, overload
-
-from typing_extensions import Final, Literal
+from typing import Any, Final, Literal, TypeVar, overload
 
 from streamlit.connections import (
     BaseConnection,
@@ -45,7 +42,7 @@ FIRST_PARTY_CONNECTIONS = {
     "sql": SQLConnection,
 }
 MODULE_EXTRACTION_REGEX = re.compile(r"No module named \'(.+)\'")
-MODULES_TO_PYPI_PACKAGES: Final[Dict[str, str]] = {
+MODULES_TO_PYPI_PACKAGES: Final[dict[str, str]] = {
     "MySQLdb": "mysqlclient",
     "psycopg2": "psycopg2-binary",
     "sqlalchemy": "sqlalchemy",
@@ -63,7 +60,7 @@ ConnectionClass = TypeVar("ConnectionClass", bound=BaseConnection[Any])
 @gather_metrics("connection")
 def _create_connection(
     name: str,
-    connection_class: Type[ConnectionClass],
+    connection_class: type[ConnectionClass],
     max_entries: int | None = None,
     ttl: float | timedelta | None = None,
     **kwargs,
@@ -77,7 +74,7 @@ def _create_connection(
     """
 
     def __create_connection(
-        name: str, connection_class: Type[ConnectionClass], **kwargs
+        name: str, connection_class: type[ConnectionClass], **kwargs
     ) -> ConnectionClass:
         return connection_class(connection_name=name, **kwargs)
 
@@ -184,7 +181,7 @@ def connection_factory(
 @overload
 def connection_factory(
     name: str,
-    type: Type[ConnectionClass],
+    type: type[ConnectionClass],
     max_entries: int | None = None,
     ttl: float | timedelta | None = None,
     **kwargs,
@@ -312,6 +309,9 @@ def connection_factory(
         if "." in connection_class:
             parts = connection_class.split(".")
             classname = parts.pop()
+
+            import importlib
+
             connection_module = importlib.import_module(".".join(parts))
             connection_class = getattr(connection_module, classname)
         else:

streamlit/runtime/forward_msg_cache.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import hashlib
-from typing import TYPE_CHECKING, Dict, List, MutableMapping, Optional
+from typing import TYPE_CHECKING, Final, MutableMapping
 from weakref import WeakKeyDictionary
 
 from streamlit import config, util
@@ -25,7 +27,7 @@ from streamlit.util import HASHLIB_KWARGS
 if TYPE_CHECKING:
     from streamlit.runtime.app_session import AppSession
 
-LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
 def populate_hash_if_needed(msg: ForwardMsg) -> str:
@@ -107,16 +109,16 @@ class ForwardMsgCache(CacheStatsProvider):
 
         """
 
-        def __init__(self, msg: Optional[ForwardMsg]):
+        def __init__(self, msg: ForwardMsg | None):
             self.msg = msg
             self._session_script_run_counts: MutableMapping[
-                "AppSession", int
+                AppSession, int
             ] = WeakKeyDictionary()
 
         def __repr__(self) -> str:
             return util.repr_(self)
 
-        def add_session_ref(self, session: "AppSession", script_run_count: int) -> None:
+        def add_session_ref(self, session: AppSession, script_run_count: int) -> None:
             """Adds a reference to a AppSession that has referenced
             this Entry's message.
 
@@ -129,18 +131,18 @@ class ForwardMsgCache(CacheStatsProvider):
             """
             prev_run_count = self._session_script_run_counts.get(session, 0)
             if script_run_count < prev_run_count:
-                LOGGER.error(
+                _LOGGER.error(
                     "New script_run_count (%s) is < prev_run_count (%s). "
                     "This should never happen!" % (script_run_count, prev_run_count)
                 )
                 script_run_count = prev_run_count
             self._session_script_run_counts[session] = script_run_count
 
-        def has_session_ref(self, session: "AppSession") -> bool:
+        def has_session_ref(self, session: AppSession) -> bool:
             return session in self._session_script_run_counts
 
         def get_session_ref_age(
-            self, session: "AppSession", script_run_count: int
+            self, session: AppSession, script_run_count: int
         ) -> int:
             """The age of the given session's reference to the Entry,
             given a new script_run_count.
@@ -148,7 +150,7 @@ class ForwardMsgCache(CacheStatsProvider):
             """
             return script_run_count - self._session_script_run_counts[session]
 
-        def remove_session_ref(self, session: "AppSession") -> None:
+        def remove_session_ref(self, session: AppSession) -> None:
             del self._session_script_run_counts[session]
 
         def has_refs(self) -> bool:
@@ -159,13 +161,13 @@ class ForwardMsgCache(CacheStatsProvider):
             return len(self._session_script_run_counts) > 0
 
     def __init__(self):
-        self._entries: Dict[str, "ForwardMsgCache.Entry"] = {}
+        self._entries: dict[str, ForwardMsgCache.Entry] = {}
 
     def __repr__(self) -> str:
         return util.repr_(self)
 
     def add_message(
-        self, msg: ForwardMsg, session: "AppSession", script_run_count: int
+        self, msg: ForwardMsg, session: AppSession, script_run_count: int
     ) -> None:
         """Add a ForwardMsg to the cache.
 
@@ -191,7 +193,7 @@ class ForwardMsgCache(CacheStatsProvider):
             self._entries[msg.hash] = entry
         entry.add_session_ref(session, script_run_count)
 
-    def get_message(self, hash: str) -> Optional[ForwardMsg]:
+    def get_message(self, hash: str) -> ForwardMsg | None:
         """Return the message with the given ID if it exists in the cache.
 
         Parameters
@@ -208,7 +210,7 @@ class ForwardMsgCache(CacheStatsProvider):
         return entry.msg if entry else None
 
     def has_message_reference(
-        self, msg: ForwardMsg, session: "AppSession", script_run_count: int
+        self, msg: ForwardMsg, session: AppSession, script_run_count: int
     ) -> bool:
         """Return True if a session has a reference to a message."""
         populate_hash_if_needed(msg)
@@ -221,7 +223,7 @@ class ForwardMsgCache(CacheStatsProvider):
         age = entry.get_session_ref_age(session, script_run_count)
         return age <= int(config.get_option("global.maxCachedMessageAge"))
 
-    def remove_refs_for_session(self, session: "AppSession") -> None:
+    def remove_refs_for_session(self, session: AppSession) -> None:
         """Remove refs for all entries for the given session.
 
         This should be called when an AppSession is disconnected or closed.
@@ -243,7 +245,7 @@ class ForwardMsgCache(CacheStatsProvider):
                 del self._entries[msg_hash]
 
     def remove_expired_entries_for_session(
-        self, session: "AppSession", script_run_count: int
+        self, session: AppSession, script_run_count: int
     ) -> None:
         """Remove any cached messages that have expired from the given session.
 
@@ -266,7 +268,7 @@ class ForwardMsgCache(CacheStatsProvider):
 
             age = entry.get_session_ref_age(session, script_run_count)
             if age > max_age:
-                LOGGER.debug(
+                _LOGGER.debug(
                     "Removing expired entry [session=%s, hash=%s, age=%s]",
                     id(session),
                     msg_hash,
@@ -282,8 +284,8 @@ class ForwardMsgCache(CacheStatsProvider):
         """Remove all entries from the cache"""
         self._entries.clear()
 
-    def get_stats(self) -> List[CacheStat]:
-        stats: List[CacheStat] = []
+    def get_stats(self) -> list[CacheStat]:
+        stats: list[CacheStat] = []
         for entry_hash, entry in self._entries.items():
             stats.append(
                 CacheStat(