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

streamlit/runtime/runtime_util.py

@@ -14,7 +14,9 @@
 
 """Runtime-related utility functions"""
 
-from typing import Any, Optional
+from __future__ import annotations
+
+from typing import Any
 
 from streamlit import config
 from streamlit.errors import MarkdownFormattedException
@@ -27,7 +29,7 @@ class MessageSizeError(MarkdownFormattedException):
 
     def __init__(self, failed_msg_str: Any):
         msg = self._get_message(failed_msg_str)
-        super(MessageSizeError, self).__init__(msg)
+        super().__init__(msg)
 
     def _get_message(self, failed_msg_str: Any) -> str:
         # This needs to have zero indentation otherwise the markdown will render incorrectly.
@@ -82,7 +84,7 @@ def serialize_forward_msg(msg: ForwardMsg) -> bytes:
 
 # This needs to be initialized lazily to avoid calling config.get_option() and
 # thus initializing config options when this file is first imported.
-_max_message_size_bytes: Optional[int] = None
+_max_message_size_bytes: int | None = None
 
 
 def get_max_message_size_bytes() -> int:

streamlit/runtime/script_data.py

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

streamlit/runtime/scriptrunner/magic.py

@@ -12,9 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import ast
+from __future__ import annotations
 
-from typing_extensions import Final
+import ast
+from typing import Any, Final
 
 from streamlit import config
 
@@ -24,7 +25,7 @@ from streamlit import config
 MAGIC_MODULE_NAME: Final = "__streamlitmagic__"
 
 
-def add_magic(code, script_path):
+def add_magic(code: str, script_path: str) -> Any:
     """Modifies the code to support magic Streamlit commands.
 
     Parameters
@@ -51,7 +52,10 @@ def add_magic(code, script_path):
 
 
 def _modify_ast_subtree(
-    tree, body_attr="body", is_root=False, file_ends_in_semicolon=False
+    tree: Any,
+    body_attr: str = "body",
+    is_root: bool = False,
+    file_ends_in_semicolon: bool = False,
 ):
     """Parses magic commands and modifies the given AST (sub)tree."""
 
@@ -118,7 +122,7 @@ def _modify_ast_subtree(
     return tree
 
 
-def _insert_import_statement(tree):
+def _insert_import_statement(tree: Any) -> None:
     """Insert Streamlit import statement at the top(ish) of the tree."""
 
     st_import = _build_st_import_statement()
@@ -226,11 +230,11 @@ def _get_st_write_from_expr(
     return st_write
 
 
-def _is_string_constant_node(node):
+def _is_string_constant_node(node) -> bool:
     return type(node) is ast.Constant and type(node.value) is str
 
 
-def _is_docstring_node(node, node_index, parent_type):
+def _is_docstring_node(node, node_index, parent_type) -> bool:
     return (
         node_index == 0
         and _is_string_constant_node(node)
@@ -238,7 +242,7 @@ def _is_docstring_node(node, node_index, parent_type):
     )
 
 
-def _does_file_end_in_semicolon(tree, code):
+def _does_file_end_in_semicolon(tree, code: str) -> bool:
     file_ends_in_semicolon = False
 
     # Avoid spending time with this operation if magic.displayLastExprIfNoSemicolon is
@@ -247,13 +251,15 @@ def _does_file_end_in_semicolon(tree, code):
         last_line_num = getattr(tree.body[-1], "end_lineno", None)
 
         if last_line_num is not None:
-            last_line_str = code.split("\n")[last_line_num - 1]
+            last_line_str: str = code.split("\n")[last_line_num - 1]
             file_ends_in_semicolon = last_line_str.strip(" ").endswith(";")
 
     return file_ends_in_semicolon
 
 
-def _is_displayable_last_expr(is_root, is_last_expr, file_ends_in_semicolon):
+def _is_displayable_last_expr(
+    is_root: bool, is_last_expr: bool, file_ends_in_semicolon: bool
+) -> bool:
     return (
         # This is a "displayable last expression" if...
         # ...it's actually the last expression...
@@ -267,5 +273,5 @@ def _is_displayable_last_expr(is_root, is_last_expr, file_ends_in_semicolon):
     )
 
 
-def _should_display_docstring_like_node_anyway(is_root):
+def _should_display_docstring_like_node_anyway(is_root: bool) -> bool:
     return config.get_option("magic.displayRootDocString") and is_root

streamlit/runtime/scriptrunner/magic_funcs.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Any
 
 from streamlit.runtime.metrics_util import gather_metrics

streamlit/runtime/scriptrunner/script_requests.py

@@ -12,10 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import threading
 from dataclasses import dataclass
 from enum import Enum
-from typing import Optional, cast
+from typing import cast
 
 from streamlit import util
 from streamlit.proto.WidgetStates_pb2 import WidgetStates
@@ -41,7 +43,7 @@ class RerunData:
     """Data attached to RERUN requests. Immutable."""
 
     query_string: str = ""
-    widget_states: Optional[WidgetStates] = None
+    widget_states: WidgetStates | None = None
     page_script_hash: str = ""
     page_name: str = ""
 
@@ -54,7 +56,7 @@ class ScriptRequest:
     """A STOP or RERUN request and associated data."""
 
     type: ScriptRequestType
-    _rerun_data: Optional[RerunData] = None
+    _rerun_data: RerunData | None = None
 
     @property
     def rerun_data(self) -> RerunData:
@@ -141,7 +143,7 @@ class ScriptRequests:
             # We'll never get here
             raise RuntimeError(f"Unrecognized ScriptRunnerState: {self._state}")
 
-    def on_scriptrunner_yield(self) -> Optional[ScriptRequest]:
+    def on_scriptrunner_yield(self) -> ScriptRequest | None:
         """Called by the ScriptRunner when it's at a yield point.
 
         If we have no request, return None.

streamlit/runtime/scriptrunner/script_run_context.py

@@ -12,14 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import collections
 import contextvars
 import threading
 from dataclasses import dataclass, field
-from typing import Callable, Counter, Dict, List, Optional, Set, Tuple
+from typing import Callable, Counter, Dict, Final, Union
 from urllib import parse
 
-from typing_extensions import Final, TypeAlias
+from typing_extensions import TypeAlias
 
 from streamlit import runtime
 from streamlit.errors import StreamlitAPIException
@@ -30,16 +32,16 @@ from streamlit.runtime.scriptrunner.script_requests import ScriptRequests
 from streamlit.runtime.state import SafeSessionState
 from streamlit.runtime.uploaded_file_manager import UploadedFileManager
 
-LOGGER: Final = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
-UserInfo: TypeAlias = Dict[str, Optional[str]]
+UserInfo: TypeAlias = Dict[str, Union[str, None]]
 
 
 # The dg_stack tracks the currently active DeltaGenerator, and is pushed to when
 # a DeltaGenerator is entered via a `with` block. This is implemented as a ContextVar
 # so that different threads or async tasks can have their own stacks.
 dg_stack: contextvars.ContextVar[
-    Tuple["streamlit.delta_generator.DeltaGenerator", ...]
+    tuple["streamlit.delta_generator.DeltaGenerator", ...]
 ] = contextvars.ContextVar("dg_stack", default=tuple())
 
 
@@ -69,15 +71,15 @@ class ScriptRunContext:
 
     gather_usage_stats: bool = False
     command_tracking_deactivated: bool = False
-    tracked_commands: List[Command] = field(default_factory=list)
+    tracked_commands: list[Command] = field(default_factory=list)
     tracked_commands_counter: Counter[str] = field(default_factory=collections.Counter)
     _set_page_config_allowed: bool = True
     _has_script_started: bool = False
-    widget_ids_this_run: Set[str] = field(default_factory=set)
-    widget_user_keys_this_run: Set[str] = field(default_factory=set)
-    form_ids_this_run: Set[str] = field(default_factory=set)
-    cursors: Dict[int, "streamlit.cursor.RunningCursor"] = field(default_factory=dict)
-    script_requests: Optional[ScriptRequests] = None
+    widget_ids_this_run: set[str] = field(default_factory=set)
+    widget_user_keys_this_run: set[str] = field(default_factory=set)
+    form_ids_this_run: set[str] = field(default_factory=set)
+    cursors: dict[int, "streamlit.cursor.RunningCursor"] = field(default_factory=dict)
+    script_requests: ScriptRequests | None = None
 
     # TODO(willhuang1997): Remove this variable when experimental query params are removed
     _experimental_query_params_used = False
@@ -153,7 +155,7 @@ SCRIPT_RUN_CONTEXT_ATTR_NAME: Final = "streamlit_script_run_ctx"
 
 
 def add_script_run_ctx(
-    thread: Optional[threading.Thread] = None, ctx: Optional[ScriptRunContext] = None
+    thread: threading.Thread | None = None, ctx: ScriptRunContext | None = None
 ):
     """Adds the current ScriptRunContext to a newly-created thread.
 
@@ -183,7 +185,7 @@ def add_script_run_ctx(
     return thread
 
 
-def get_script_run_ctx(suppress_warning: bool = False) -> Optional[ScriptRunContext]:
+def get_script_run_ctx(suppress_warning: bool = False) -> ScriptRunContext | None:
     """
     Parameters
     ----------
@@ -196,15 +198,13 @@ def get_script_run_ctx(suppress_warning: bool = False) -> Optional[ScriptRunCont
 
     """
     thread = threading.current_thread()
-    ctx: Optional[ScriptRunContext] = getattr(
-        thread, SCRIPT_RUN_CONTEXT_ATTR_NAME, None
-    )
+    ctx: ScriptRunContext | None = getattr(thread, SCRIPT_RUN_CONTEXT_ATTR_NAME, None)
     if ctx is None and runtime.exists() and not suppress_warning:
         # Only warn about a missing ScriptRunContext if suppress_warning is False, and
         # we were started via `streamlit run`. Otherwise, the user is likely running a
         # script "bare", and doesn't need to be warned about streamlit
         # bits that are irrelevant when not connected to a session.
-        LOGGER.warning("Thread '%s': missing ScriptRunContext", thread.name)
+        _LOGGER.warning("Thread '%s': missing ScriptRunContext", thread.name)
 
     return ctx
 

streamlit/runtime/scriptrunner/script_runner.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import gc
 import sys
 import threading
@@ -19,7 +21,7 @@ import types
 from contextlib import contextmanager
 from enum import Enum
 from timeit import default_timer as timer
-from typing import Callable, Dict, Optional
+from typing import Callable, Final
 
 from blinker import Signal
 
@@ -47,7 +49,7 @@ from streamlit.runtime.state import (
 from streamlit.runtime.uploaded_file_manager import UploadedFileManager
 from streamlit.vendor.ipython.modified_sys_path import modified_sys_path
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
 class ScriptRunnerEvent(Enum):
@@ -103,7 +105,7 @@ class ScriptRunner:
         uploaded_file_mgr: UploadedFileManager,
         script_cache: ScriptCache,
         initial_rerun_data: RerunData,
-        user_info: Dict[str, Optional[str]],
+        user_info: dict[str, str | None],
     ):
         """Initialize the ScriptRunner.
 
@@ -181,7 +183,7 @@ class ScriptRunner:
         self._execing = False
 
         # This is initialized in start()
-        self._script_thread: Optional[threading.Thread] = None
+        self._script_thread: threading.Thread | None = None
 
     def __repr__(self) -> str:
         return util.repr_(self)
@@ -493,7 +495,7 @@ class ScriptRunner:
 
             # This will be set to a RerunData instance if our execution
             # is interrupted by a RerunException.
-            rerun_exception_data: Optional[RerunData] = None
+            rerun_exception_data: RerunData | None = None
 
             # If the script stops early, we don't want to remove unseen widgets,
             # so we track this to potentially skip session state cleanup later.

streamlit/runtime/secrets.py

@@ -19,12 +19,10 @@ import threading
 from copy import deepcopy
 from typing import (
     Any,
-    Dict,
     Final,
     ItemsView,
     Iterator,
     KeysView,
-    List,
     Mapping,
     NoReturn,
     ValuesView,
@@ -37,8 +35,8 @@ import streamlit.watcher.path_watcher
 from streamlit import file_util, runtime
 from streamlit.logger import get_logger
 
-_LOGGER = get_logger(__name__)
-SECRETS_FILE_LOCS: Final[List[str]] = [
+_LOGGER: Final = get_logger(__name__)
+SECRETS_FILE_LOCS: Final[list[str]] = [
     file_util.get_streamlit_file_path("secrets.toml"),
     # NOTE: The order here is important! Project-level secrets should overwrite global
     # secrets.
@@ -107,7 +105,7 @@ class AttrDict(Mapping[str, Any]):
     def __setattr__(self, key, value) -> NoReturn:
         raise TypeError("Secrets does not support attribute assignment.")
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         return deepcopy(self.__nested_secrets__)
 
 
@@ -118,7 +116,7 @@ class Secrets(Mapping[str, Any]):
     Safe to use from multiple threads.
     """
 
-    def __init__(self, file_paths: List[str]):
+    def __init__(self, file_paths: list[str]):
         # Our secrets dict.
         self._secrets: Mapping[str, Any] | None = None
         self._lock = threading.RLock()

streamlit/runtime/session_manager.py

@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from abc import abstractmethod
 from dataclasses import dataclass
-from typing import Callable, Dict, List, Optional, cast
-
-from typing_extensions import Protocol
+from typing import Callable, Protocol, cast
 
 from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
 from streamlit.runtime.app_session import AppSession
@@ -64,7 +64,7 @@ class SessionInfo:
     the ForwardMsgCache.
     """
 
-    client: Optional[SessionClient]
+    client: SessionClient | None
     session: AppSession
     script_run_count: int = 0
 
@@ -90,7 +90,7 @@ class SessionStorageError(Exception):
 
 class SessionStorage(Protocol):
     @abstractmethod
-    def get(self, session_id: str) -> Optional[SessionInfo]:
+    def get(self, session_id: str) -> SessionInfo | None:
         """Return the SessionInfo corresponding to session_id, or None if one does not
         exist.
 
@@ -152,7 +152,7 @@ class SessionStorage(Protocol):
         raise NotImplementedError
 
     @abstractmethod
-    def list(self) -> List[SessionInfo]:
+    def list(self) -> list[SessionInfo]:
         """List all sessions tracked by this SessionStorage.
 
         Returns
@@ -210,7 +210,7 @@ class SessionManager(Protocol):
         session_storage: SessionStorage,
         uploaded_file_manager: UploadedFileManager,
         script_cache: ScriptCache,
-        message_enqueued_callback: Optional[Callable[[], None]],
+        message_enqueued_callback: Callable[[], None] | None,
     ) -> None:
         """Initialize a SessionManager with the given SessionStorage.
 
@@ -235,9 +235,9 @@ class SessionManager(Protocol):
         self,
         client: SessionClient,
         script_data: ScriptData,
-        user_info: Dict[str, Optional[str]],
-        existing_session_id: Optional[str] = None,
-        session_id_override: Optional[str] = None,
+        user_info: dict[str, str | None],
+        existing_session_id: str | None = None,
+        session_id_override: str | None = None,
     ) -> str:
         """Create a new session or connect to an existing one.
 
@@ -290,7 +290,7 @@ class SessionManager(Protocol):
         raise NotImplementedError
 
     @abstractmethod
-    def get_session_info(self, session_id: str) -> Optional[SessionInfo]:
+    def get_session_info(self, session_id: str) -> SessionInfo | None:
         """Return the SessionInfo for the given id, or None if no such session
         exists.
 
@@ -306,7 +306,7 @@ class SessionManager(Protocol):
         raise NotImplementedError
 
     @abstractmethod
-    def list_sessions(self) -> List[SessionInfo]:
+    def list_sessions(self) -> list[SessionInfo]:
         """Return the SessionInfo for all sessions managed by this SessionManager.
 
         Returns
@@ -344,7 +344,7 @@ class SessionManager(Protocol):
         """
         self.close_session(session_id)
 
-    def get_active_session_info(self, session_id: str) -> Optional[ActiveSessionInfo]:
+    def get_active_session_info(self, session_id: str) -> ActiveSessionInfo | None:
         """Return the ActiveSessionInfo for the given id, or None if either no such
         session exists or the session is not active.
 
@@ -371,7 +371,7 @@ class SessionManager(Protocol):
         """
         return self.get_active_session_info(session_id) is not None
 
-    def list_active_sessions(self) -> List[ActiveSessionInfo]:
+    def list_active_sessions(self) -> list[ActiveSessionInfo]:
         """Return the session info for all active sessions tracked by this SessionManager.
 
         Returns

streamlit/runtime/state/common.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Functions and data structures shared by session_state.py and widgets.py"""
+
 from __future__ import annotations
 
 import hashlib
@@ -23,8 +24,8 @@ from typing import (
     Any,
     Callable,
     Dict,
+    Final,
     Generic,
-    Optional,
     Sequence,
     Tuple,
     TypeVar,
@@ -32,7 +33,7 @@ from typing import (
 )
 
 from google.protobuf.message import Message
-from typing_extensions import Final, TypeAlias
+from typing_extensions import TypeAlias
 
 from streamlit import util
 from streamlit.errors import StreamlitAPIException
@@ -151,7 +152,7 @@ class RegisterWidgetResult(Generic[T_co]):
     @classmethod
     def failure(
         cls, deserializer: WidgetDeserializer[T_co]
-    ) -> "RegisterWidgetResult[T_co]":
+    ) -> RegisterWidgetResult[T_co]:
         """The canonical way to construct a RegisterWidgetResult in cases
         where the true widget value could not be determined.
         """
@@ -200,7 +201,7 @@ def compute_widget_id(
     return f"{GENERATED_WIDGET_ID_PREFIX}-{h.hexdigest()}-{user_key}"
 
 
-def user_key_from_widget_id(widget_id: str) -> Optional[str]:
+def user_key_from_widget_id(widget_id: str) -> str | None:
     """Return the user key portion of a widget id, or None if the id does not
     have a user key.
 

streamlit/runtime/state/query_params.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from dataclasses import dataclass, field
-from typing import Dict, Iterable, Iterator, List, MutableMapping, Union
+from typing import Iterable, Iterator, MutableMapping
 from urllib import parse
 
 from streamlit.constants import EMBED_QUERY_PARAMS_KEYS
@@ -27,7 +29,7 @@ class QueryParams(MutableMapping[str, str]):
     It stores str keys with str and List[str] values.
     """
 
-    _query_params: Dict[str, Union[List[str], str]] = field(default_factory=dict)
+    _query_params: dict[str, list[str] | str] = field(default_factory=dict)
 
     def __iter__(self) -> Iterator[str]:
         self._ensure_single_query_api_used()
@@ -59,7 +61,7 @@ class QueryParams(MutableMapping[str, str]):
         except KeyError:
             raise KeyError(missing_key_error_message(key))
 
-    def __setitem__(self, key: str, value: Union[str, Iterable[str]]) -> None:
+    def __setitem__(self, key: str, value: str | Iterable[str]) -> None:
         if isinstance(value, dict):
             raise StreamlitAPIException(
                 f"You cannot set a query params key `{key}` to a dictionary."
@@ -86,7 +88,7 @@ class QueryParams(MutableMapping[str, str]):
         except KeyError:
             raise KeyError(missing_key_error_message(key))
 
-    def get_all(self, key: str) -> List[str]:
+    def get_all(self, key: str) -> list[str]:
         self._ensure_single_query_api_used()
         if key not in self._query_params or key in EMBED_QUERY_PARAMS_KEYS:
             return []
@@ -128,7 +130,7 @@ class QueryParams(MutableMapping[str, str]):
 
         self._send_query_param_msg()
 
-    def to_dict(self) -> Dict[str, str]:
+    def to_dict(self) -> dict[str, str]:
         self._ensure_single_query_api_used()
         # return the last query param if multiple values are set
         return {
@@ -137,7 +139,7 @@ class QueryParams(MutableMapping[str, str]):
             if key not in EMBED_QUERY_PARAMS_KEYS
         }
 
-    def set_with_no_forward_msg(self, key: str, val: Union[List[str], str]) -> None:
+    def set_with_no_forward_msg(self, key: str, val: list[str] | str) -> None:
         self._query_params[key] = val
 
     def clear_with_no_forward_msg(self) -> None:

streamlit/runtime/state/query_params_proxy.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Dict, Iterable, Iterator, List, MutableMapping, Union
+from __future__ import annotations
+
+from typing import Iterable, Iterator, MutableMapping
 
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.state.query_params import missing_key_error_message
@@ -47,7 +49,7 @@ class QueryParamsProxy(MutableMapping[str, str]):
             del qp[key]
 
     @gather_metrics("query_params.set_item")
-    def __setitem__(self, key: str, value: Union[str, Iterable[str]]) -> None:
+    def __setitem__(self, key: str, value: str | Iterable[str]) -> None:
         with get_session_state().query_params() as qp:
             qp[key] = value
 
@@ -67,12 +69,12 @@ class QueryParamsProxy(MutableMapping[str, str]):
                 raise AttributeError(missing_key_error_message(key))
 
     @gather_metrics("query_params.set_attr")
-    def __setattr__(self, key: str, value: Union[str, Iterable[str]]) -> None:
+    def __setattr__(self, key: str, value: str | Iterable[str]) -> None:
         with get_session_state().query_params() as qp:
             qp[key] = value
 
     @gather_metrics("query_params.get_all")
-    def get_all(self, key: str) -> List[str]:
+    def get_all(self, key: str) -> list[str]:
         """
         Get a list of all query parameter values associated to a given key.
 
@@ -107,7 +109,7 @@ class QueryParamsProxy(MutableMapping[str, str]):
             qp.clear()
 
     @gather_metrics("query_params.to_dict")
-    def to_dict(self) -> Dict[str, str]:
+    def to_dict(self) -> dict[str, str]:
         """
         Get all query parameters as a dictionary.
 

streamlit/runtime/state/safe_session_state.py

@@ -12,9 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import threading
 from contextlib import contextmanager
-from typing import Any, Callable, Dict, Iterator, List, Optional, Set
+from typing import Any, Callable, Iterator
 
 from streamlit.proto.WidgetStates_pb2 import WidgetState as WidgetStateProto
 from streamlit.proto.WidgetStates_pb2 import WidgetStates as WidgetStatesProto
@@ -47,7 +49,7 @@ class SafeSessionState:
         object.__setattr__(self, "_yield_callback", yield_callback)
 
     def register_widget(
-        self, metadata: WidgetMetadata[T], user_key: Optional[str]
+        self, metadata: WidgetMetadata[T], user_key: str | None
     ) -> RegisterWidgetResult[T]:
         self._yield_callback()
         with self._lock:
@@ -62,7 +64,7 @@ class SafeSessionState:
             #  to a Lock.)
             self._state.on_script_will_rerun(latest_widget_states)
 
-    def on_script_finished(self, widget_ids_this_run: Set[str]) -> None:
+    def on_script_finished(self, widget_ids_this_run: set[str]) -> None:
         with self._lock:
             self._state.on_script_finished(widget_ids_this_run)
 
@@ -70,7 +72,7 @@ class SafeSessionState:
         with self._lock:
             self._state.maybe_check_serializable()
 
-    def get_widget_states(self) -> List[WidgetStateProto]:
+    def get_widget_states(self) -> list[WidgetStateProto]:
         """Return a list of serialized widget values for each widget with a value."""
         with self._lock:
             return self._state.get_widget_states()
@@ -80,7 +82,7 @@ class SafeSessionState:
             return self._state.is_new_state_value(user_key)
 
     @property
-    def filtered_state(self) -> Dict[str, Any]:
+    def filtered_state(self) -> dict[str, Any]:
         """The combined session and widget state, excluding keyless widgets."""
         with self._lock:
             return self._state.filtered_state