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

streamlit/cursor.py

@@ -12,15 +12,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, List, Optional, Tuple
+from __future__ import annotations
+
+from typing import Any
 
 from streamlit import util
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 
 
 def make_delta_path(
-    root_container: int, parent_path: Tuple[int, ...], index: int
-) -> List[int]:
+    root_container: int, parent_path: tuple[int, ...], index: int
+) -> list[int]:
     delta_path = [root_container]
     delta_path.extend(parent_path)
     delta_path.append(index)
@@ -28,8 +30,8 @@ def make_delta_path(
 
 
 def get_container_cursor(
-    root_container: Optional[int],
-) -> Optional["RunningCursor"]:
+    root_container: int | None,
+) -> RunningCursor | None:
     """Return the top-level RunningCursor for the given container.
     This is the cursor that is used when user code calls something like
     `st.foo` (which uses the main container) or `st.sidebar.foo` (which uses
@@ -69,7 +71,7 @@ class Cursor:
         raise NotImplementedError()
 
     @property
-    def parent_path(self) -> Tuple[int, ...]:
+    def parent_path(self) -> tuple[int, ...]:
         """The cursor's parent's path within its container."""
         raise NotImplementedError()
 
@@ -79,7 +81,7 @@ class Cursor:
         raise NotImplementedError()
 
     @property
-    def delta_path(self) -> List[int]:
+    def delta_path(self) -> list[int]:
         """The complete path of the delta pointed to by this cursor - its
         container, parent path, and index.
         """
@@ -89,7 +91,7 @@ class Cursor:
     def is_locked(self) -> bool:
         raise NotImplementedError()
 
-    def get_locked_cursor(self, **props) -> "LockedCursor":
+    def get_locked_cursor(self, **props) -> LockedCursor:
         raise NotImplementedError()
 
     @property
@@ -103,7 +105,7 @@ class Cursor:
 
 
 class RunningCursor(Cursor):
-    def __init__(self, root_container: int, parent_path: Tuple[int, ...] = ()):
+    def __init__(self, root_container: int, parent_path: tuple[int, ...] = ()):
         """A moving pointer to a delta location in the app.
 
         RunningCursors auto-increment to the next available location when you
@@ -127,7 +129,7 @@ class RunningCursor(Cursor):
         return self._root_container
 
     @property
-    def parent_path(self) -> Tuple[int, ...]:
+    def parent_path(self) -> tuple[int, ...]:
         return self._parent_path
 
     @property
@@ -138,7 +140,7 @@ class RunningCursor(Cursor):
     def is_locked(self) -> bool:
         return False
 
-    def get_locked_cursor(self, **props) -> "LockedCursor":
+    def get_locked_cursor(self, **props) -> LockedCursor:
         locked_cursor = LockedCursor(
             root_container=self._root_container,
             parent_path=self._parent_path,
@@ -155,7 +157,7 @@ class LockedCursor(Cursor):
     def __init__(
         self,
         root_container: int,
-        parent_path: Tuple[int, ...] = (),
+        parent_path: tuple[int, ...] = (),
         index: int = 0,
         **props,
     ):
@@ -188,7 +190,7 @@ class LockedCursor(Cursor):
         return self._root_container
 
     @property
-    def parent_path(self) -> Tuple[int, ...]:
+    def parent_path(self) -> tuple[int, ...]:
         return self._parent_path
 
     @property
@@ -199,7 +201,7 @@ class LockedCursor(Cursor):
     def is_locked(self) -> bool:
         return True
 
-    def get_locked_cursor(self, **props) -> "LockedCursor":
+    def get_locked_cursor(self, **props) -> LockedCursor:
         self._props = props
         return self
 

streamlit/delta_generator.py

@@ -27,7 +27,6 @@ from typing import (
     Iterable,
     Literal,
     NoReturn,
-    Type,
     TypeVar,
     cast,
     overload,
@@ -340,19 +339,17 @@ class DeltaGenerator(
             if name in streamlit_methods:
                 if self._root_container == RootContainer.SIDEBAR:
                     message = (
-                        "Method `%(name)s()` does not exist for "
-                        "`st.sidebar`. Did you mean `st.%(name)s()`?" % {"name": name}
+                        f"Method `{name}()` does not exist for "
+                        f"`st.sidebar`. Did you mean `st.{name}()`?"
                     )
                 else:
                     message = (
-                        "Method `%(name)s()` does not exist for "
+                        f"Method `{name}()` does not exist for "
                         "`DeltaGenerator` objects. Did you mean "
-                        "`st.%(name)s()`?" % {"name": name}
+                        "`st.{name}()`?"
                     )
             else:
-                message = "`%(name)s()` is not a valid Streamlit command." % {
-                    "name": name
-                }
+                message = f"`{name}()` is not a valid Streamlit command."
 
             raise StreamlitAPIException(message)
 
@@ -432,7 +429,7 @@ class DeltaGenerator(
         self,
         delta_type: str,
         element_proto: Message,
-        return_value: Type[NoValue],
+        return_value: type[NoValue],
         add_rows_metadata: AddRowsMetadata | None = None,
         element_width: int | None = None,
         element_height: int | None = None,
@@ -468,7 +465,7 @@ class DeltaGenerator(
         self,
         delta_type: str,
         element_proto: Message,
-        return_value: Type[NoValue] | Value | None = None,
+        return_value: type[NoValue] | Value | None = None,
         add_rows_metadata: AddRowsMetadata | None = None,
         element_width: int | None = None,
         element_height: int | None = None,
@@ -479,7 +476,7 @@ class DeltaGenerator(
         self,
         delta_type: str,
         element_proto: Message,
-        return_value: Type[NoValue] | Value | None = None,
+        return_value: type[NoValue] | Value | None = None,
         add_rows_metadata: AddRowsMetadata | None = None,
         element_width: int | None = None,
         element_height: int | None = None,
@@ -837,7 +834,7 @@ def _value_or_dg(value: None, dg: DG) -> DG:
 
 
 @overload
-def _value_or_dg(value: Type[NoValue], dg: DG) -> None:  # type: ignore[misc]
+def _value_or_dg(value: type[NoValue], dg: DG) -> None:  # type: ignore[misc]
     ...
 
 
@@ -855,7 +852,7 @@ def _value_or_dg(value: Value, dg: DG) -> Value:
 
 
 def _value_or_dg(
-    value: Type[NoValue] | Value | None,
+    value: type[NoValue] | Value | None,
     dg: DG,
 ) -> DG | Value | None:
     """Return either value, or None, or dg.

streamlit/deprecation_util.py

@@ -11,16 +11,17 @@
 # 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 functools
-from typing import Any, Callable, List, TypeVar, cast
+from typing import Any, Callable, Final, TypeVar, cast
 
 import streamlit
 from streamlit import config
 from streamlit.logger import get_logger
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 TFunc = TypeVar("TFunc", bound=Callable[..., Any])
 TObj = TypeVar("TObj", bound=object)
@@ -187,7 +188,7 @@ def _create_deprecated_obj_wrapper(obj: TObj, show_warning: Callable[[], Any]) -
             return getattr(obj, attr)
 
         @staticmethod
-        def _get_magic_functions(cls) -> List[str]:
+        def _get_magic_functions(cls) -> list[str]:
             # ignore the handful of magic functions we cannot override without
             # breaking the Wrapper.
             ignore = ("__class__", "__dict__", "__getattribute__", "__getattr__")

streamlit/echo.py

@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import ast
 import contextlib
 import re
 import textwrap
 import traceback
-from typing import Any, Dict, Iterable, Optional
+from typing import Any, Iterable
 
 from streamlit.runtime.metrics_util import gather_metrics
 
@@ -66,7 +68,7 @@ def echo(code_location="above"):
 
         # Use ast to parse the Python file and find the code block to display
         root_node = ast.parse("".join(source_lines))
-        line_to_node_map: Dict[int, Any] = {}
+        line_to_node_map: dict[int, Any] = {}
 
         def collect_body_statements(node: ast.AST) -> None:
             if not hasattr(node, "body"):
@@ -109,7 +111,7 @@ def _get_initial_indent(lines: Iterable[str]) -> int:
     return 0
 
 
-def _get_indent(line: str) -> Optional[int]:
+def _get_indent(line: str) -> int | None:
     """Get the number of whitespaces at the beginning of the given line.
     If the line is empty, or if it contains just whitespace and a newline,
     return None.

streamlit/elements/alert.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.Alert_pb2 import Alert as AlertProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -27,10 +29,10 @@ class AlertMixin:
     @gather_metrics("error")
     def error(
         self,
-        body: "SupportsStr",
+        body: SupportsStr,
         *,  # keyword-only args:
-        icon: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        icon: str | None = None,
+    ) -> DeltaGenerator:
         """Display error message.
 
         Parameters
@@ -59,10 +61,10 @@ class AlertMixin:
     @gather_metrics("warning")
     def warning(
         self,
-        body: "SupportsStr",
+        body: SupportsStr,
         *,  # keyword-only args:
-        icon: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        icon: str | None = None,
+    ) -> DeltaGenerator:
         """Display warning message.
 
         Parameters
@@ -91,10 +93,10 @@ class AlertMixin:
     @gather_metrics("info")
     def info(
         self,
-        body: "SupportsStr",
+        body: SupportsStr,
         *,  # keyword-only args:
-        icon: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        icon: str | None = None,
+    ) -> DeltaGenerator:
         """Display an informational message.
 
         Parameters
@@ -124,10 +126,10 @@ class AlertMixin:
     @gather_metrics("success")
     def success(
         self,
-        body: "SupportsStr",
+        body: SupportsStr,
         *,  # keyword-only args:
-        icon: Optional[str] = None,
-    ) -> "DeltaGenerator":
+        icon: str | None = None,
+    ) -> DeltaGenerator:
         """Display a success message.
 
         Parameters
@@ -154,6 +156,6 @@ class AlertMixin:
         return self.dg._enqueue("alert", alert_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)

streamlit/elements/altair_utils.py

@@ -17,22 +17,24 @@
 These classes are used to pass some important info to add_rows.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
-from typing import Hashable, List, Optional, TypedDict
+from typing import Hashable, TypedDict
 
 
 class PrepDataColumns(TypedDict):
     """Columns used for the prep_data step in Altair Arrow charts."""
 
-    x_column: Optional[str]
-    y_column_list: List[str]
-    color_column: Optional[str]
-    size_column: Optional[str]
+    x_column: str | None
+    y_column_list: list[str]
+    color_column: str | None
+    size_column: str | None
 
 
 @dataclass
 class AddRowsMetadata:
     """Metadata needed by add_rows on native charts."""
 
-    last_index: Optional[Hashable]
+    last_index: Hashable | None
     columns: PrepDataColumns

streamlit/elements/arrow.py

@@ -64,7 +64,7 @@ class ArrowMixin:
         hide_index: bool | None = None,
         column_order: Iterable[str] | None = None,
         column_config: ColumnConfigMappingInput | None = None,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a dataframe as an interactive table.
 
         This command works with dataframes from Pandas, PyArrow, Snowpark, and PySpark.
@@ -239,7 +239,7 @@ class ArrowMixin:
         return self.dg._enqueue("arrow_data_frame", proto)
 
     @gather_metrics("table")
-    def table(self, data: Data = None) -> "DeltaGenerator":
+    def table(self, data: Data = None) -> DeltaGenerator:
         """Display a static table.
 
         This differs from ``st.dataframe`` in that the table in this case is
@@ -284,7 +284,7 @@ class ArrowMixin:
         return self.dg._enqueue("arrow_table", proto)
 
     @gather_metrics("add_rows")
-    def add_rows(self, data: "Data" = None, **kwargs) -> "DeltaGenerator" | None:
+    def add_rows(self, data: Data = None, **kwargs) -> DeltaGenerator | None:
         """Concatenate a dataframe to the bottom of the current one.
 
         Parameters
@@ -338,7 +338,7 @@ class ArrowMixin:
         return self.dg._arrow_add_rows(data, **kwargs)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 

streamlit/elements/arrow_altair.py

@@ -21,17 +21,7 @@ from __future__ import annotations
 from contextlib import nullcontext
 from datetime import date
 from enum import Enum
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Collection,
-    Dict,
-    List,
-    Literal,
-    Sequence,
-    Tuple,
-    cast,
-)
+from typing import TYPE_CHECKING, Any, Collection, Literal, Sequence, cast
 
 import streamlit.elements.arrow_vega_lite as arrow_vega_lite
 from streamlit import type_util
@@ -108,7 +98,7 @@ class ArrowAltairMixin:
         *,
         x: str | None = None,
         y: str | Sequence[str] | None = None,
-        color: str | Color | List[Color] | None = None,
+        color: str | Color | list[Color] | None = None,
         width: int = 0,
         height: int = 0,
         use_container_width: bool = True,
@@ -266,7 +256,7 @@ class ArrowAltairMixin:
         *,
         x: str | None = None,
         y: str | Sequence[str] | None = None,
-        color: str | Color | List[Color] | None = None,
+        color: str | Color | list[Color] | None = None,
         width: int = 0,
         height: int = 0,
         use_container_width: bool = True,
@@ -425,7 +415,7 @@ class ArrowAltairMixin:
         *,
         x: str | None = None,
         y: str | Sequence[str] | None = None,
-        color: str | Color | List[Color] | None = None,
+        color: str | Color | list[Color] | None = None,
         width: int = 0,
         height: int = 0,
         use_container_width: bool = True,
@@ -586,12 +576,12 @@ class ArrowAltairMixin:
         *,
         x: str | None = None,
         y: str | Sequence[str] | None = None,
-        color: str | Color | List[Color] | None = None,
+        color: str | Color | list[Color] | None = None,
         size: str | float | int | None = None,
         width: int = 0,
         height: int = 0,
         use_container_width: bool = True,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a scatterplot chart.
 
         This is syntax-sugar around ``st.altair_chart``. The main difference
@@ -849,8 +839,8 @@ def _is_date_column(df: pd.DataFrame, name: str | None) -> bool:
 
 def _melt_data(
     df: pd.DataFrame,
-    columns_to_leave_alone: List[str],
-    columns_to_melt: List[str] | None,
+    columns_to_leave_alone: list[str],
+    columns_to_melt: list[str] | None,
     new_y_column_name: str,
     new_color_column_name: str,
 ) -> pd.DataFrame:
@@ -893,10 +883,10 @@ def _melt_data(
 def prep_data(
     df: pd.DataFrame,
     x_column: str | None,
-    y_column_list: List[str],
+    y_column_list: list[str],
     color_column: str | None,
     size_column: str | None,
-) -> Tuple[pd.DataFrame, str | None, str | None, str | None, str | None]:
+) -> tuple[pd.DataFrame, str | None, str | None, str | None, str | None]:
     """Prepares the data for charting. This is also used in add_rows.
 
     Returns the prepared dataframe and the new names of the x column (taking the index reset into
@@ -939,11 +929,11 @@ def _generate_chart(
     data: Data | None,
     x_from_user: str | None = None,
     y_from_user: str | Sequence[str] | None = None,
-    color_from_user: str | Color | List[Color] | None = None,
+    color_from_user: str | Color | list[Color] | None = None,
     size_from_user: str | float | None = None,
     width: int = 0,
     height: int = 0,
-) -> Tuple[alt.Chart, AddRowsMetadata]:
+) -> tuple[alt.Chart, AddRowsMetadata]:
     """Function to use the chart's type, data columns and indices to figure out the chart's spec."""
     import altair as alt
 
@@ -1028,7 +1018,7 @@ def _generate_chart(
 
 
 def _maybe_reset_index_in_place(
-    df: pd.DataFrame, x_column: str | None, y_column_list: List[str]
+    df: pd.DataFrame, x_column: str | None, y_column_list: list[str]
 ) -> str | None:
     if x_column is None and len(y_column_list) > 0:
         if df.index.name is None:
@@ -1087,10 +1077,10 @@ def _maybe_convert_color_column_in_place(df: pd.DataFrame, color_column: str | N
 def _convert_col_names_to_str_in_place(
     df: pd.DataFrame,
     x_column: str | None,
-    y_column_list: List[str],
+    y_column_list: list[str],
     color_column: str | None,
     size_column: str | None,
-) -> Tuple[str | None, List[str], str | None, str | None]:
+) -> tuple[str | None, list[str], str | None, str | None]:
     """Converts column names to strings, since Vega-Lite does not accept ints, etc."""
     import pandas as pd
 
@@ -1108,7 +1098,7 @@ def _convert_col_names_to_str_in_place(
 
 def _parse_generic_column(
     df: pd.DataFrame, column_or_value: Any
-) -> Tuple[str | None, Any]:
+) -> tuple[str | None, Any]:
     if isinstance(column_or_value, str) and column_or_value in df.columns:
         column_name = column_or_value
         value = None
@@ -1141,8 +1131,8 @@ def _parse_y_columns(
     df: pd.DataFrame,
     y_from_user: str | Sequence[str] | None,
     x_column: str | None,
-) -> List[str]:
-    y_column_list: List[str] = []
+) -> list[str]:
+    y_column_list: list[str] = []
 
     if y_from_user is None:
         y_column_list = list(df.columns)
@@ -1210,10 +1200,10 @@ def _get_axis_config(df: pd.DataFrame, column_name: str | None, grid: bool) -> a
 def _maybe_melt(
     df: pd.DataFrame,
     x_column: str | None,
-    y_column_list: List[str],
+    y_column_list: list[str],
     color_column: str | None,
     size_column: str | None,
-) -> Tuple[pd.DataFrame, str | None, str | None]:
+) -> tuple[pd.DataFrame, str | None, str | None]:
     """If multiple columns are set for y, melt the dataframe into long format."""
     y_column: str | None
 
@@ -1324,8 +1314,8 @@ def _get_color_encoding(
     df: pd.DataFrame,
     color_value: Color | None,
     color_column: str | None,
-    y_column_list: List[str],
-    color_from_user: str | Color | List[Color] | None,
+    y_column_list: list[str],
+    color_from_user: str | Color | list[Color] | None,
 ) -> alt.Color | alt.ColorValue | None:
     import altair as alt
 
@@ -1362,7 +1352,7 @@ def _get_color_encoding(
         raise StreamlitInvalidColorError(df, color_from_user)
 
     elif color_column is not None:
-        column_type: str | Tuple[str, List[Any]]
+        column_type: str | tuple[str, list[Any]]
 
         if color_column == MELTED_COLOR_COLUMN_NAME:
             column_type = "nominal"
@@ -1525,7 +1515,7 @@ def marshall(
 
     datasets = {}
 
-    def id_transform(data) -> Dict[str, str]:
+    def id_transform(data) -> dict[str, str]:
         """Altair data transformer that returns a fake named dataset with the
         object id.
         """

streamlit/elements/arrow_vega_lite.py

@@ -13,18 +13,16 @@
 # limitations under the License.
 
 """A Python wrapper around Vega-Lite."""
+
 from __future__ import annotations
 
 import json
-from typing import TYPE_CHECKING, Any, Dict, Optional, Union, cast
-
-from typing_extensions import Final, Literal
+from typing import TYPE_CHECKING, Any, Literal, cast
 
 import streamlit.elements.lib.dicttools as dicttools
 from streamlit.elements import arrow
 from streamlit.elements.arrow import Data
 from streamlit.errors import StreamlitAPIException
-from streamlit.logger import get_logger
 from streamlit.proto.ArrowVegaLiteChart_pb2 import (
     ArrowVegaLiteChart as ArrowVegaLiteChartProto,
 )
@@ -34,19 +32,16 @@ if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
 
-LOGGER: Final = get_logger(__name__)
-
-
 class ArrowVegaLiteMixin:
     @gather_metrics("vega_lite_chart")
     def vega_lite_chart(
         self,
         data: Data = None,
-        spec: Dict[str, Any] | None = None,
+        spec: dict[str, Any] | None = None,
         use_container_width: bool = False,
         theme: Literal["streamlit"] | None = "streamlit",
         **kwargs: Any,
-    ) -> "DeltaGenerator":
+    ) -> DeltaGenerator:
         """Display a chart using the Vega-Lite library.
 
         Parameters
@@ -117,7 +112,7 @@ class ArrowVegaLiteMixin:
         return self.dg._enqueue("arrow_vega_lite_chart", proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)
 
@@ -125,9 +120,9 @@ class ArrowVegaLiteMixin:
 def marshall(
     proto: ArrowVegaLiteChartProto,
     data: Data = None,
-    spec: Optional[Dict[str, Any]] = None,
+    spec: dict[str, Any] | None = None,
     use_container_width: bool = False,
-    theme: Union[None, Literal["streamlit"]] = "streamlit",
+    theme: None | Literal["streamlit"] = "streamlit",
     **kwargs,
 ):
     """Construct a Vega-Lite chart object.
@@ -203,8 +198,8 @@ _CHANNELS = {
     "x2",
     "y2",
     "xError",
-    "yError2",
-    "xError",
+    "xError2",
+    "yError",
     "yError2",
     "longitude",
     "latitude",

streamlit/elements/balloons.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.Balloons_pb2 import Balloons as BalloonsProto
@@ -23,7 +25,7 @@ if TYPE_CHECKING:
 
 class BalloonsMixin:
     @gather_metrics("balloons")
-    def balloons(self) -> "DeltaGenerator":
+    def balloons(self) -> DeltaGenerator:
         """Draw celebratory balloons.
 
         Example
@@ -40,6 +42,6 @@ class BalloonsMixin:
         return self.dg._enqueue("balloons", balloons_proto)
 
     @property
-    def dg(self) -> "DeltaGenerator":
+    def dg(self) -> DeltaGenerator:
         """Get our DeltaGenerator."""
         return cast("DeltaGenerator", self)