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

streamlit/elements/bokeh_chart.py

@@ -14,11 +14,11 @@
 
 """A Python wrapper around Bokeh."""
 
+from __future__ import annotations
+
 import hashlib
 import json
-from typing import TYPE_CHECKING, cast
-
-from typing_extensions import Final
+from typing import TYPE_CHECKING, Final, cast
 
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.BokehChart_pb2 import BokehChart as BokehChartProto
@@ -37,9 +37,9 @@ class BokehMixin:
     @gather_metrics("bokeh_chart")
     def bokeh_chart(
         self,
-        figure: "Figure",
+        figure: Figure,
         use_container_width: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display an interactive Bokeh chart.
 
         Bokeh is a charting library for Python. The arguments to this function
@@ -99,14 +99,14 @@ class BokehMixin:
         return self.dg._enqueue("bokeh_chart", bokeh_chart_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
 
 def marshall(
     proto: BokehChartProto,
-    figure: "Figure",
+    figure: Figure,
     use_container_width: bool,
     element_id: str,
 ) -> None:

streamlit/elements/code.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING, Optional, cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
 
 from streamlit.proto.Code_pb2 import Code as CodeProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -28,9 +30,9 @@ class CodeMixin:
     def code(
         self,
         body: SupportsStr,
-        language: Optional[str] = "python",
+        language: str | None = "python",
         line_numbers: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a code block with optional syntax highlighting.
 
         Parameters
@@ -66,6 +68,6 @@ class CodeMixin:
         return self.dg._enqueue("code", code_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/deck_gl_json_chart.py

@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import hashlib
 import json
-from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, cast
-
-from typing_extensions import Final
+from typing import TYPE_CHECKING, Any, Final, Mapping, cast
 
 from streamlit import config
 from streamlit.proto.DeckGlJsonChart_pb2 import DeckGlJsonChart as PydeckProto
@@ -39,9 +39,9 @@ class PydeckMixin:
     @gather_metrics("pydeck_chart")
     def pydeck_chart(
         self,
-        pydeck_obj: Optional["Deck"] = None,
+        pydeck_obj: Deck | None = None,
         use_container_width: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Draw a chart using the PyDeck library.
 
         This supports 3D maps, point clouds, and more! More info about PyDeck
@@ -131,12 +131,12 @@ class PydeckMixin:
         return self.dg._enqueue("deck_gl_json_chart", pydeck_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
 
-def _get_pydeck_tooltip(pydeck_obj: Optional["Deck"]) -> Optional[Dict[str, str]]:
+def _get_pydeck_tooltip(pydeck_obj: Deck | None) -> dict[str, str] | None:
     if pydeck_obj is None:
         return None
 
@@ -156,7 +156,7 @@ def _get_pydeck_tooltip(pydeck_obj: Optional["Deck"]) -> Optional[Dict[str, str]
 
 def marshall(
     pydeck_proto: PydeckProto,
-    pydeck_obj: Optional["Deck"],
+    pydeck_obj: Deck | None,
     use_container_width: bool,
 ) -> None:
     if pydeck_obj is None:

streamlit/elements/doc_string.py

@@ -14,17 +14,16 @@
 
 """Allows us to create and absorb changes (aka Deltas) to elements."""
 
+from __future__ import annotations
+
 import ast
 import contextlib
 import inspect
 import re
 import types
-from typing import TYPE_CHECKING, Any, cast
-
-from typing_extensions import Final
+from typing import TYPE_CHECKING, Any, Final, cast
 
 import streamlit
-from streamlit.logger import get_logger
 from streamlit.proto.DocString_pb2 import DocString as DocStringProto
 from streamlit.proto.DocString_pb2 import Member as MemberProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -38,15 +37,12 @@ if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
 
-LOGGER: Final = get_logger(__name__)
-
-
 CONFUSING_STREAMLIT_SIG_PREFIXES: Final = ("(element, ",)
 
 
 class HelpMixin:
     @gather_metrics("help")
-    def help(self, obj: Any = streamlit) -> "DeltaGenerator":
+    def help(self, obj: Any = streamlit) -> DeltaGenerator:
         """Display help and other information for a given object.
 
         Depending on the type of object that is passed in, this displays the
@@ -123,7 +119,7 @@ class HelpMixin:
         return self.dg._enqueue("doc_string", doc_string_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 

streamlit/elements/empty.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 TYPE_CHECKING, cast
 
 from streamlit.proto.Empty_pb2 import Empty as EmptyProto
@@ -21,7 +23,7 @@ if TYPE_CHECKING:
 
 
 class EmptyMixin:
-    def empty(self) -> "DeltaGenerator":
+    def empty(self) -> DeltaGenerator:
         """Insert a single-element container.
 
         Inserts a container into your app that can be used to hold a single element.
@@ -70,6 +72,6 @@ class EmptyMixin:
         return self.dg._enqueue("empty", empty_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/exception.py

@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import os
 import traceback
-from typing import TYPE_CHECKING, List, Optional, cast
-
-from typing_extensions import Final
+from typing import TYPE_CHECKING, Final, cast
 
 import streamlit
 from streamlit.errors import (
@@ -33,7 +33,7 @@ from streamlit.runtime.metrics_util import gather_metrics
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
-LOGGER: Final = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 # When client.showErrorDetails is False, we show a generic warning in the
 # frontend when we encounter an uncaught app exception.
@@ -48,7 +48,7 @@ _STREAMLIT_DIR: Final = os.path.join(
 
 class ExceptionMixin:
     @gather_metrics("exception")
-    def exception(self, exception: BaseException) -> "DeltaGenerator":
+    def exception(self, exception: BaseException) -> DeltaGenerator:
         """Display an exception.
 
         Parameters
@@ -69,7 +69,7 @@ class ExceptionMixin:
         return self.dg._enqueue("exception", exception_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
@@ -124,7 +124,7 @@ def marshall(exception_proto: ExceptionProto, exception: BaseException) -> None:
         # Sometimes the exception's __str__/__unicode__ method itself
         # raises an error.
         exception_proto.message = ""
-        LOGGER.warning(
+        _LOGGER.warning(
             """
 
 Streamlit was unable to parse the data from an exception in the user's script.
@@ -193,7 +193,7 @@ def _format_syntax_error_message(exception: SyntaxError) -> str:
 
 def _get_stack_trace_str_list(
     exception: BaseException, strip_streamlit_stack_entries: bool = False
-) -> List[str]:
+) -> list[str]:
     """Get the stack trace for the given exception.
 
     Parameters
@@ -213,7 +213,7 @@ def _get_stack_trace_str_list(
         The exception traceback as a list of strings
 
     """
-    extracted_traceback: Optional[traceback.StackSummary] = None
+    extracted_traceback: traceback.StackSummary | None = None
     if isinstance(exception, StreamlitAPIWarning):
         extracted_traceback = exception.tacked_on_stack
     elif hasattr(exception, "__traceback__"):
@@ -253,7 +253,7 @@ def _is_in_streamlit_package(file: str) -> bool:
 
 def _get_nonstreamlit_traceback(
     extracted_tb: traceback.StackSummary,
-) -> List[traceback.FrameSummary]:
+) -> list[traceback.FrameSummary]:
     return [
         entry for entry in extracted_tb if not _is_in_streamlit_package(entry.filename)
     ]

streamlit/elements/form.py

@@ -14,9 +14,7 @@
 from __future__ import annotations
 
 import textwrap
-from typing import TYPE_CHECKING, NamedTuple, cast
-
-from typing_extensions import Literal
+from typing import TYPE_CHECKING, Literal, NamedTuple, cast
 
 from streamlit import runtime
 from streamlit.errors import StreamlitAPIException

streamlit/elements/graphviz_chart.py

@@ -14,14 +14,15 @@
 
 """Streamlit support for GraphViz charts."""
 
+from __future__ import annotations
+
 import hashlib
 from typing import TYPE_CHECKING, Union, cast
 
-from typing_extensions import Final, TypeAlias
+from typing_extensions import TypeAlias
 
 from streamlit import type_util
 from streamlit.errors import StreamlitAPIException
-from streamlit.logger import get_logger
 from streamlit.proto.GraphVizChart_pb2 import GraphVizChart as GraphVizChartProto
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.util import HASHLIB_KWARGS
@@ -31,8 +32,6 @@ if TYPE_CHECKING:
 
     from streamlit.delta_generator import DeltaGenerator
 
-LOGGER: Final = get_logger(__name__)
-
 FigureOrDot: TypeAlias = Union["graphviz.Graph", "graphviz.Digraph", str]
 
 
@@ -42,7 +41,7 @@ class GraphvizMixin:
         self,
         figure_or_dot: FigureOrDot,
         use_container_width: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a graph using the dagre-d3 library.
 
         Parameters
@@ -113,7 +112,7 @@ class GraphvizMixin:
         return self.dg._enqueue("graphviz_chart", graphviz_chart_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 

streamlit/elements/heading.py

@@ -12,10 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from enum import Enum
-from typing import TYPE_CHECKING, Optional, Union, cast
+from typing import TYPE_CHECKING, Literal, Union, cast
 
-from typing_extensions import Literal
+from typing_extensions import TypeAlias
 
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Heading_pb2 import Heading as HeadingProto
@@ -33,8 +35,8 @@ class HeadingProtoTag(Enum):
     SUBHEADER_TAG = "h3"
 
 
-Anchor = Optional[Union[str, Literal[False]]]
-Divider = Optional[Union[bool, str]]
+Anchor: TypeAlias = Union[str, Literal[False], None]
+Divider: TypeAlias = Union[bool, str, None]
 
 
 class HeadingMixin:
@@ -44,9 +46,9 @@ class HeadingMixin:
         body: SupportsStr,
         anchor: Anchor = None,
         *,  # keyword-only arguments:
-        help: Optional[str] = None,
+        help: str | None = None,
         divider: Divider = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display text in header formatting.
 
         Parameters
@@ -114,9 +116,9 @@ class HeadingMixin:
         body: SupportsStr,
         anchor: Anchor = None,
         *,  # keyword-only arguments:
-        help: Optional[str] = None,
+        help: str | None = None,
         divider: Divider = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display text in subheader formatting.
 
         Parameters
@@ -184,8 +186,8 @@ class HeadingMixin:
         body: SupportsStr,
         anchor: Anchor = None,
         *,  # keyword-only arguments:
-        help: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        help: str | None = None,
+    ) -> DeltaGenerator:
         """Display text in title formatting.
 
         Each document should have a single `st.title()`, although this is not
@@ -239,12 +241,12 @@ class HeadingMixin:
         )
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
     @staticmethod
-    def _handle_divider_color(divider):
+    def _handle_divider_color(divider: Divider) -> str:
         if divider is True:
             return "auto"
         valid_colors = [
@@ -258,7 +260,7 @@ class HeadingMixin:
             "rainbow",
         ]
         if divider in valid_colors:
-            return divider
+            return cast(str, divider)
         else:
             raise StreamlitAPIException(
                 f"Divider parameter has invalid value: `{divider}`. Please choose from: {', '.join(valid_colors)}."
@@ -269,7 +271,7 @@ class HeadingMixin:
         tag: HeadingProtoTag,
         body: SupportsStr,
         anchor: Anchor = None,
-        help: Optional[str] = None,
+        help: str | None = None,
         divider: Divider = False,
     ) -> HeadingProto:
         proto = HeadingProto()

streamlit/elements/iframe.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING, Optional, cast
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
 
 from streamlit.proto.IFrame_pb2 import IFrame as IFrameProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -26,10 +28,10 @@ class IframeMixin:
     def _iframe(
         self,
         src: str,
-        width: Optional[int] = None,
-        height: Optional[int] = None,
+        width: int | None = None,
+        height: int | None = None,
         scrolling: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Load a remote URL in an iframe.
 
         Parameters
@@ -60,10 +62,10 @@ class IframeMixin:
     def _html(
         self,
         html: str,
-        width: Optional[int] = None,
-        height: Optional[int] = None,
+        width: int | None = None,
+        height: int | None = None,
         scrolling: bool = False,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display an HTML string in an iframe.
 
         Parameters
@@ -91,17 +93,17 @@ class IframeMixin:
         return self.dg._enqueue("iframe", iframe_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
 
 def marshall(
     proto: IFrameProto,
-    src: Optional[str] = None,
-    srcdoc: Optional[str] = None,
-    width: Optional[int] = None,
-    height: Optional[int] = None,
+    src: str | None = None,
+    srcdoc: str | None = None,
+    width: int | None = None,
+    height: int | None = None,
     scrolling: bool = False,
 ) -> None:
     """Marshalls data into an IFrame proto.

streamlit/elements/image.py

@@ -87,13 +87,13 @@ class ImageMixin:
         image: ImageOrImageList,
         # TODO: Narrow type of caption, dependent on type of image,
         #  by way of overload
-        caption: str | List[str] | None = None,
+        caption: str | list[str] | None = None,
         width: int | None = None,
         use_column_width: UseColumnWith = None,
         clamp: bool = False,
         channels: Channels = "RGB",
         output_format: ImageFormatOrAuto = "auto",
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display an image or list of images.
 
         Parameters
@@ -171,7 +171,7 @@ class ImageMixin:
         return self.dg._enqueue("imgs", image_list_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
@@ -243,7 +243,7 @@ def _BytesIO_to_bytes(data: io.BytesIO) -> bytes:
     return data.getvalue()
 
 
-def _np_array_to_bytes(array: "npt.NDArray[Any]", output_format="JPEG") -> bytes:
+def _np_array_to_bytes(array: npt.NDArray[Any], output_format: str = "JPEG") -> bytes:
     import numpy as np
     from PIL import Image
 
@@ -253,11 +253,11 @@ def _np_array_to_bytes(array: "npt.NDArray[Any]", output_format="JPEG") -> bytes
     return _PIL_to_bytes(img, format)
 
 
-def _4d_to_list_3d(array: "npt.NDArray[Any]") -> List["npt.NDArray[Any]"]:
+def _4d_to_list_3d(array: npt.NDArray[Any]) -> list[npt.NDArray[Any]]:
     return [array[i, :, :, :] for i in range(0, array.shape[0])]
 
 
-def _verify_np_shape(array: "npt.NDArray[Any]") -> "npt.NDArray[Any]":
+def _verify_np_shape(array: npt.NDArray[Any]) -> npt.NDArray[Any]:
     if len(array.shape) not in (2, 3):
         raise StreamlitAPIException("Numpy shape has to be of length 2 or 3.")
     if len(array.shape) == 3 and array.shape[-1] not in (1, 3, 4):
@@ -307,7 +307,7 @@ def _ensure_image_size_and_format(
     return image_data
 
 
-def _clip_image(image: "npt.NDArray[Any]", clamp: bool) -> "npt.NDArray[Any]":
+def _clip_image(image: npt.NDArray[Any], clamp: bool) -> npt.NDArray[Any]:
     import numpy as np
 
     data = image
@@ -451,7 +451,7 @@ def image_to_url(
 def marshall_images(
     coordinates: str,
     image: ImageOrImageList,
-    caption: str | "npt.NDArray[Any]" | List[str] | None,
+    caption: str | npt.NDArray[Any] | list[str] | None,
     width: int | WidthBehaviour,
     proto_imgs: ImageListProto,
     clamp: bool,

streamlit/elements/json.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import json
-from typing import TYPE_CHECKING, Any, List, Union, cast
+from typing import TYPE_CHECKING, Any, cast
 
 from streamlit.proto.Json_pb2 import Json as JsonProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -24,7 +26,7 @@ if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
 
-def _ensure_serialization(o: object) -> Union[str, List[Any]]:
+def _ensure_serialization(o: object) -> str | list[Any]:
     """A repr function for json.dumps default arg, which tries to serialize sets as lists"""
     if isinstance(o, set):
         return list(o)
@@ -38,7 +40,7 @@ class JsonMixin:
         body: object,
         *,  # keyword-only arguments:
         expanded: bool = True,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display object or string as a pretty-printed JSON string.
 
         Parameters
@@ -94,6 +96,6 @@ class JsonMixin:
         return self.dg._enqueue("json", json_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/layouts.py

@@ -14,7 +14,9 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, List, Literal, Optional, Sequence, Union, cast
+from typing import TYPE_CHECKING, Literal, Sequence, Union, cast
+
+from typing_extensions import TypeAlias
 
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Block_pb2 import Block as BlockProto
@@ -24,14 +26,14 @@ if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
     from streamlit.elements.lib.mutable_status_container import StatusContainer
 
-SpecType = Union[int, Sequence[Union[int, float]]]
+SpecType: TypeAlias = Union[int, Sequence[Union[int, float]]]
 
 
 class LayoutsMixin:
     @gather_metrics("container")
     def container(
         self, *, height: int | None = None, border: bool | None = None
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Insert a multi-element container.
 
         Inserts an invisible container into your app that can be used to hold
@@ -142,8 +144,8 @@ class LayoutsMixin:
     # TODO: Enforce that columns are not nested or in Sidebar
     @gather_metrics("columns")
     def columns(
-        self, spec: SpecType, *, gap: Optional[str] = "small"
-    ) -> List["DeltaGenerator"]:
+        self, spec: SpecType, *, gap: str | None = "small"
+    ) -> list[DeltaGenerator]:
         """Insert containers laid out as side-by-side columns.
 
         Inserts a number of multi-element containers laid out side-by-side and
@@ -268,7 +270,7 @@ class LayoutsMixin:
         return [row._block(column_proto(w / total_weight)) for w in weights]
 
     @gather_metrics("tabs")
-    def tabs(self, tabs: Sequence[str]) -> Sequence["DeltaGenerator"]:
+    def tabs(self, tabs: Sequence[str]) -> Sequence[DeltaGenerator]:
         r"""Insert containers separated into tabs.
 
         Inserts a number of multi-element containers as tabs.
@@ -380,7 +382,7 @@ class LayoutsMixin:
         return tuple(tab_container._block(tab_proto(tab_label)) for tab_label in tabs)
 
     @gather_metrics("expander")
-    def expander(self, label: str, expanded: bool = False) -> "DeltaGenerator":
+    def expander(self, label: str, expanded: bool = False) -> DeltaGenerator:
         r"""Insert a multi-element container that can be expanded/collapsed.
 
         Inserts a container into your app that can be used to hold multiple elements
@@ -481,8 +483,8 @@ class LayoutsMixin:
         *,
         expanded: bool = False,
         state: Literal["running", "complete", "error"] = "running",
-    ) -> "StatusContainer":
-        """Insert a status container to display output from long-running tasks.
+    ) -> StatusContainer:
+        r"""Insert a status container to display output from long-running tasks.
 
         Inserts a container into your app that is typically used to show the status and
         details of a process or task. The container can hold multiple elements and can
@@ -597,6 +599,6 @@ class LayoutsMixin:
         )
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/lib/column_config_utils.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 import json
 from enum import Enum
-from typing import TYPE_CHECKING, Dict, Final, List, Literal, Mapping, Union
+from typing import TYPE_CHECKING, Dict, Final, Literal, Mapping, Union
 
 from typing_extensions import TypeAlias
 
@@ -69,7 +69,7 @@ DataframeSchema: TypeAlias = Dict[str, ColumnDataKind]
 
 # This mapping contains all editable column types mapped to the data kinds
 # that the column type is compatible for editing.
-_EDITING_COMPATIBILITY_MAPPING: Final[Dict[ColumnType, List[ColumnDataKind]]] = {
+_EDITING_COMPATIBILITY_MAPPING: Final[dict[ColumnType, list[ColumnDataKind]]] = {
     "text": [ColumnDataKind.STRING, ColumnDataKind.EMPTY],
     "number": [
         ColumnDataKind.INTEGER,