streamlit-nightly 1.45.2.dev20250512__py3-none-any.whl → 1.45.2.dev20250513__py3-none-any.whl

streamlit/__init__.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 # isort: skip_file
+# ruff: noqa: E402, PLC0414, A001
 
 """Streamlit.
 

streamlit/auth_util.py

@@ -32,21 +32,21 @@ if TYPE_CHECKING:
 class AuthCache:
     """Simple cache implementation for storing info required for Authlib."""
 
-    def __init__(self):
-        self.cache = {}
+    def __init__(self) -> None:
+        self.cache: dict[str, Any] = {}
 
-    def get(self, key):
+    def get(self, key: str) -> Any:
         return self.cache.get(key)
 
     # for set method, we are follow the same signature used in Authlib
     # the expires_in is not used in our case
-    def set(self, key, value, expires_in):  # noqa: ARG002
+    def set(self, key: str, value: Any, expires_in: int | None = None) -> None:  # noqa: ARG002
         self.cache[key] = value
 
-    def get_dict(self):
+    def get_dict(self) -> dict[str, Any]:
         return self.cache
 
-    def delete(self, key):
+    def delete(self, key: str) -> None:
         self.cache.pop(key, None)
 
 

streamlit/commands/echo.py

@@ -19,12 +19,12 @@ import contextlib
 import re
 import textwrap
 import traceback
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Literal
 
 from streamlit.runtime.metrics_util import gather_metrics
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable
+    from collections.abc import Generator, Iterable
 
 _SPACES_RE = re.compile("\\s*")
 _EMPTY_LINE_RE = re.compile("\\s*\n")
@@ -32,7 +32,9 @@ _EMPTY_LINE_RE = re.compile("\\s*\n")
 
 @gather_metrics("echo")
 @contextlib.contextmanager
-def echo(code_location="above"):
+def echo(
+    code_location: Literal["above", "below"] = "above",
+) -> Generator[None, None, None]:
     """Use in a `with` block to draw some code on the app, then execute it.
 
     Parameters

streamlit/components/v1/custom_component.py

@@ -183,7 +183,7 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
             # other arguments change, and the component's iframe won't be
             # remounted on the frontend.
 
-            def marshall_element_args():
+            def marshall_element_args() -> None:
                 element.component_instance.json_args = serialized_json_args
                 element.component_instance.special_args.extend(special_args)
 
@@ -210,7 +210,7 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
                 )
             element.component_instance.id = computed_id
 
-            def deserialize_component(ui_value):
+            def deserialize_component(ui_value: Any) -> Any:
                 # ui_value is an object from json, an ArrowTable proto, or a bytearray
                 return ui_value
 

streamlit/config.py

@@ -1651,7 +1651,7 @@ def _set_development_mode() -> None:
 
 def on_config_parsed(
     func: Callable[[], None], force_connect: bool = False, lock: bool = False
-) -> Callable[[], bool]:
+) -> Callable[[], None]:
     """Wait for the config file to be parsed then call func.
 
     If the config file has already been parsed, just calls func immediately
@@ -1671,7 +1671,7 @@ def on_config_parsed(
 
     Returns
     -------
-    Callable[[], bool]
+    Callable[[], None]
         A function that the caller can use to deregister func.
     """
 
@@ -1680,13 +1680,13 @@ def on_config_parsed(
     # leading to a memory leak because the Signal will keep a reference of the
     # callable argument. When the callable argument is an object method, then
     # the reference to that object won't be released.
-    def receiver(_):
-        return func_with_lock()
+    def receiver(_: Any) -> None:
+        func_with_lock()
 
-    def disconnect():
-        return _on_config_parsed.disconnect(receiver)
+    def disconnect() -> None:
+        _on_config_parsed.disconnect(receiver)
 
-    def func_with_lock():
+    def func_with_lock() -> None:
         if lock:
             with _config_lock:
                 func()

streamlit/config_util.py

@@ -54,16 +54,16 @@ def show_config(
         )
     )
 
-    def append_desc(text):
+    def append_desc(text: str) -> None:
         out.append("# " + cli_util.style_for_cli(text, bold=True))
 
-    def append_comment(text):
+    def append_comment(text: str) -> None:
         out.append("# " + cli_util.style_for_cli(text))
 
-    def append_section(text):
+    def append_section(text: str) -> None:
         out.append(cli_util.style_for_cli(text, bold=True, fg="green"))
 
-    def append_setting(text):
+    def append_setting(text: str) -> None:
         out.append(cli_util.style_for_cli(text, fg="green"))
 
     for section in section_descriptions:
@@ -178,8 +178,7 @@ def _clean_paragraphs(txt: str) -> list[str]:
     # Strip both leading and trailing newlines.
     txt = txt.strip("\n")
     paragraphs = txt.split("\n\n")
-    cleaned_paragraphs = [
+    return [
         "\n".join(_clean(line) for line in paragraph.split("\n"))
         for paragraph in paragraphs
     ]
-    return cleaned_paragraphs

streamlit/delta_generator.py

@@ -522,9 +522,12 @@ class DeltaGenerator(
 
     def _block(
         self,
-        block_proto: Block_pb2.Block = Block_pb2.Block(),
+        block_proto: Block_pb2.Block | None = None,
         dg_type: type | None = None,
     ) -> DeltaGenerator:
+        if block_proto is None:
+            block_proto = Block_pb2.Block()
+
         # Operate on the active DeltaGenerator, in case we're in a `with` block.
         dg = self._active_dg
 

streamlit/deprecation_util.py

@@ -93,7 +93,7 @@ def deprecate_func_name(
     """
 
     @functools.wraps(func)
-    def wrapped_func(*args, **kwargs):
+    def wrapped_func(*args: Any, **kwargs: Any) -> Any:
         result = func(*args, **kwargs)
         show_deprecation_warning(
             make_deprecated_name_warning(
@@ -167,7 +167,7 @@ def _create_deprecated_obj_wrapper(obj: TObj, show_warning: Callable[[], Any]) -
             show_warning()
 
     class Wrapper:
-        def __init__(self):
+        def __init__(self) -> None:
             # Override all the Wrapped object's magic functions
             for name in Wrapper._get_magic_functions(obj.__class__):
                 setattr(
@@ -176,7 +176,7 @@ def _create_deprecated_obj_wrapper(obj: TObj, show_warning: Callable[[], Any]) -
                     property(self._make_magic_function_proxy(name)),
                 )
 
-        def __getattr__(self, attr):
+        def __getattr__(self, attr: str) -> Any:
             # We handle __getattr__ separately from our other magic
             # functions. The wrapped class may not actually implement it,
             # but we still need to implement it to call all its normal
@@ -199,8 +199,8 @@ def _create_deprecated_obj_wrapper(obj: TObj, show_warning: Callable[[], Any]) -
             ]
 
         @staticmethod
-        def _make_magic_function_proxy(name):
-            def proxy(self, *args):  # noqa: ARG001
+        def _make_magic_function_proxy(name: str) -> Callable[[Any], Any]:
+            def proxy(_self: Any, *args: Any) -> Any:
                 maybe_show_warning()
                 return getattr(obj, name)
 

streamlit/elements/doc_string.py

@@ -283,9 +283,7 @@ def _get_variable_name_from_code_str(code: str) -> str | None:
     ):
         # A common pattern is to add "," at the end of a magic command to make it print.
         # This removes that final ",", so it looks nicer.
-        code = code.removesuffix(",")
-
-        return code
+        return code.removesuffix(",")
 
     arg_node = _get_stcommand_arg(tree)
 
@@ -427,7 +425,7 @@ def _get_weight(value: Any) -> int:
     return 0
 
 
-def _get_value(obj, var_name):
+def _get_value(obj: object, var_name: str | None) -> str | None:
     obj_value = _get_human_readable_value(obj)
 
     if obj_value is not None:
@@ -459,7 +457,7 @@ def _get_value(obj, var_name):
     return obj_value
 
 
-def _get_human_readable_value(value):
+def _get_human_readable_value(value: Any) -> str | None:
     if isinstance(value, Secrets):
         # Don't want to read secrets.toml because that will show a warning if there's no
         # secrets.toml file.
@@ -508,7 +506,7 @@ def _is_computed_property(obj: object, attr_name: str) -> bool:
     return False
 
 
-def _get_members(obj):
+def _get_members(obj: object) -> list[MemberProto]:
     members_for_sorting = []
 
     for attr_name in dir(obj):

streamlit/elements/lib/built_in_chart_utils.py

@@ -954,7 +954,9 @@ def _get_color_encoding(
         # If the color value is color-like, return that.
         if is_color_like(cast("Any", color_value)):
             if len(y_column_list) != 1:
-                raise StreamlitColorLengthError([color_value], y_column_list)
+                raise StreamlitColorLengthError(
+                    [color_value] if color_value else [], y_column_list
+                )
 
             return alt.ColorValue(to_css_color(cast("Any", color_value)))
 
@@ -975,7 +977,7 @@ def _get_color_encoding(
                 title=" ",
             )
 
-        raise StreamlitInvalidColorError(df, color_from_user)
+        raise StreamlitInvalidColorError(color_from_user)
 
     if color_column is not None:
         column_type: VegaLiteType
@@ -1132,7 +1134,7 @@ def _get_y_encoding_type(
 
 
 class StreamlitColumnNotFoundError(StreamlitAPIException):
-    def __init__(self, df, col_name, *args):
+    def __init__(self, df: pd.DataFrame, col_name: str, *args: Any) -> None:
         available_columns = ", ".join(str(c) for c in list(df.columns))
         message = (
             f'Data does not have a column named `"{col_name}"`. '
@@ -1142,8 +1144,7 @@ class StreamlitColumnNotFoundError(StreamlitAPIException):
 
 
 class StreamlitInvalidColorError(StreamlitAPIException):
-    def __init__(self, df, color_from_user, *args):
-        ", ".join(str(c) for c in list(df.columns))
+    def __init__(self, color_from_user: str | Color | list[Color] | None) -> None:
         message = f"""
 This does not look like a valid color argument: `{color_from_user}`.
 
@@ -1156,14 +1157,18 @@ The color argument can be:
 * The name of a column.
 * Or a list of colors, matching the number of y columns to draw.
         """
-        super().__init__(message, *args)
+        super().__init__(message)
 
 
 class StreamlitColorLengthError(StreamlitAPIException):
-    def __init__(self, color_values, y_column_list, *args):
+    def __init__(
+        self,
+        color_values: str | Color | Collection[Color] | None,
+        y_column_list: list[str],
+    ) -> None:
         message = (
             f"The list of colors `{color_values}` must have the same "
             "length as the list of columns to be colored "
             f"`{y_column_list}`."
         )
-        super().__init__(message, *args)
+        super().__init__(message)

streamlit/elements/lib/policies.py

@@ -97,7 +97,7 @@ def check_session_state_rules(
 
 
 class CachedWidgetWarning(StreamlitAPIWarning):
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             """
 Your script uses a widget command in a cached function

streamlit/elements/spinner.py

@@ -83,7 +83,7 @@ def spinner(
 
     try:
 
-        def set_message():
+        def set_message() -> None:
             with display_message_lock:
                 if display_message:
                     spinner_proto = SpinnerProto()

streamlit/elements/vega_charts.py

@@ -360,7 +360,7 @@ def _convert_altair_to_vega_lite_spec(
         datasets[name] = data_bytes
         return {"name": name}
 
-    alt.data_transformers.register("id", id_transform)  # type: ignore[attr-defined,unused-ignore]
+    alt.data_transformers.register("id", id_transform)  # type: ignore[arg-type,attr-defined,unused-ignore]
 
     # The default altair theme has some width/height defaults defined
     # which are not useful for Streamlit. Therefore, we change the theme to

streamlit/elements/widgets/slider.py

@@ -692,7 +692,7 @@ class SliderMixin:
         if single_value:
             value = [value]
 
-        def value_to_generic_type(v):
+        def value_to_generic_type(v: Any) -> SliderProto.DataType.ValueType:
             if isinstance(v, Integral):
                 return SUPPORTED_TYPES[Integral]
             if isinstance(v, Real):

streamlit/elements/write.py

@@ -156,7 +156,7 @@ class WriteMixin:
         streamed_response: str = ""
         written_content: list[Any] = StreamingOutput()
 
-        def flush_stream_response():
+        def flush_stream_response() -> None:
             """Write the full response to the app."""
             nonlocal streamed_response
             nonlocal stream_container
@@ -428,7 +428,7 @@ class WriteMixin:
                 "when called as `st.write()` or `st.sidebar.write()`."
             )
 
-        def flush_buffer():
+        def flush_buffer() -> None:
             if string_buffer:
                 text_content = " ".join(string_buffer)
                 # The usage of empty here prevents

streamlit/error_util.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Final
+from typing import Any, Final
 
 import streamlit
 from streamlit import config
@@ -26,17 +26,24 @@ _LOGGER: Final = get_logger(__name__)
 
 
 def _print_rich_exception(e: BaseException) -> None:
-    from rich import box, panel
+    from rich.box import Box
+    from rich.panel import Panel
 
     # Monkey patch the panel to use our custom box style
-    class ConfigurablePanel(panel.Panel):
+    class ConfigurablePanel(Panel):
         def __init__(
             self,
-            renderable,
-            box=box.Box("────\n    \n────\n    \n────\n────\n    \n────\n"),
-            **kwargs,
-        ):
-            super().__init__(renderable, box, **kwargs)
+            renderable: Any,
+            box: Box | None = None,
+            **kwargs: Any,
+        ) -> None:
+            super().__init__(
+                renderable,
+                box
+                if box is not None
+                else Box("────\n    \n────\n    \n────\n────\n    \n────\n"),
+                **kwargs,
+            )
 
     from rich import traceback as rich_traceback
 

streamlit/errors.py

@@ -20,6 +20,7 @@ from typing import TYPE_CHECKING, Any, Literal
 from streamlit import util
 
 if TYPE_CHECKING:
+    from collections.abc import Collection
     from datetime import date, time
 
 
@@ -140,7 +141,7 @@ class StreamlitAPIWarning(StreamlitAPIException, Warning):
     instead.
     """
 
-    def __init__(self, *args):
+    def __init__(self, *args: Any) -> None:
         super().__init__(*args)
         import inspect
         import traceback
@@ -157,7 +158,7 @@ class StreamlitModuleNotFoundError(StreamlitAPIWarning):
     that is not one of our core dependencies.
     """
 
-    def __init__(self, module_name, *args):
+    def __init__(self, module_name: str, *args: Any) -> None:
         message = (
             f'This Streamlit command requires module "{module_name}" to be installed.'
         )
@@ -220,7 +221,7 @@ class StreamlitInvalidURLError(LocalizableStreamlitException):
 class StreamlitInvalidColumnSpecError(LocalizableStreamlitException):
     """Exception raised when no weights are specified, or a negative weight is specified."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "The `spec` argument to `st.columns` must be either a "
             "positive integer (number of columns) or a list of positive numbers (width ratios of the columns). "
@@ -370,7 +371,7 @@ class StreamlitInvalidNumberFormatError(LocalizableStreamlitException):
 class StreamlitMissingPageLabelError(LocalizableStreamlitException):
     """Exception raised when a page_link is created without a label."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "The `label` param is required for external links used with `st.page_link` - please provide a `label`."
         )
@@ -409,7 +410,7 @@ class StreamlitPageNotFoundError(LocalizableStreamlitException):
 class StreamlitFragmentWidgetsNotAllowedOutsideError(LocalizableStreamlitException):
     """Exception raised when the fragment attempts to write to an element outside of its container."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__("Fragments cannot write widgets to outside containers.")
 
 
@@ -418,7 +419,7 @@ class StreamlitInvalidFormCallbackError(LocalizableStreamlitException):
     the `st.form_submit_button`.
     """
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__(
             "Within a form, callbacks can only be defined on `st.form_submit_button`. "
             "Defining callbacks on other widgets inside a form is not allowed."
@@ -436,7 +437,9 @@ class StreamlitValueAssignmentNotAllowedError(LocalizableStreamlitException):
 
 
 class StreamlitInvalidColorError(LocalizableStreamlitException):
-    def __init__(self, color):
+    def __init__(
+        self, color: str | Collection[Any] | tuple[int, int, int, int]
+    ) -> None:
         super().__init__(
             "This does not look like a valid color: {color}.\n\n"
             "Colors must be in one of the following formats:"

streamlit/file_util.py

@@ -19,11 +19,14 @@ import errno
 import io
 import os
 from pathlib import Path
-from typing import Final
+from typing import IO, TYPE_CHECKING, Any, Final
 
 from streamlit import env_util, errors
 from streamlit.string_util import is_binary_string
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 # Configuration and credentials are stored inside the ~/.streamlit folder
 CONFIG_FOLDER_NAME: Final = ".streamlit"
 
@@ -63,7 +66,7 @@ def get_encoded_file_data(
 
 
 @contextlib.contextmanager
-def streamlit_read(path, binary=False):
+def streamlit_read(path: str, binary: bool = False) -> Generator[IO[Any], None, None]:
     """Opens a context to read this file relative to the streamlit path.
 
     For example:
@@ -88,7 +91,7 @@ def streamlit_read(path, binary=False):
 
 
 @contextlib.contextmanager
-def streamlit_write(path, binary=False):
+def streamlit_write(path: str, binary: bool = False) -> Generator[IO[Any], None, None]:
     """Opens a file for writing within the streamlit path, and
     ensuring that the path exists.
 

streamlit/git_util.py

@@ -16,10 +16,13 @@ from __future__ import annotations
 
 import os
 import re
-from typing import Any
+from typing import TYPE_CHECKING, cast
 
 from streamlit import util
 
+if TYPE_CHECKING:
+    from git import Commit, Remote, RemoteReference, Repo
+
 # Github has two URLs, one that is https and one that is ssh
 GITHUB_HTTP_URL = r"^https://(www\.)?github.com/(.+)/(.+)(?:.git)?$"
 GITHUB_SSH_URL = r"^git@github.com:(.+)/(.+)(?:.git)?$"
@@ -31,7 +34,9 @@ MIN_GIT_VERSION = (2, 7, 0)
 
 
 class GitRepo:
-    def __init__(self, path):
+    repo: Repo | None
+
+    def __init__(self, path: str) -> None:
         # If we have a valid repo, git_version will be a tuple
         # of 3+ ints: (major, minor, patch, possible_additional_patch_number)
         self.git_version: tuple[int, ...] | None = None
@@ -39,11 +44,7 @@ class GitRepo:
         try:
             import git
 
-            # GitPython is not fully typed, and mypy is outputting inconsistent
-            # type errors on Mac and Linux. We bypass type checking entirely
-            # by redeclaring the `git` import as an "Any".
-            git_package: Any = git
-            self.repo = git_package.Repo(path, search_parent_directories=True)
+            self.repo = git.Repo(path, search_parent_directories=True)
             self.git_version = self.repo.git.version_info
 
             if self.git_version >= MIN_GIT_VERSION:
@@ -69,8 +70,8 @@ class GitRepo:
         )
 
     @property
-    def tracking_branch(self):
-        if not self.is_valid():
+    def tracking_branch(self) -> RemoteReference | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         if self.is_head_detached:
@@ -79,41 +80,45 @@ class GitRepo:
         return self.repo.active_branch.tracking_branch()
 
     @property
-    def untracked_files(self):
-        if not self.is_valid():
+    def untracked_files(self) -> list[str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         return self.repo.untracked_files
 
     @property
-    def is_head_detached(self):
-        if not self.is_valid():
+    def is_head_detached(self) -> bool:
+        if self.repo is None or not self.is_valid():
             return False
 
         return self.repo.head.is_detached
 
     @property
-    def uncommitted_files(self):
-        if not self.is_valid():
+    def uncommitted_files(self) -> list[str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
-        return [item.a_path for item in self.repo.index.diff(None)]
+        return [cast("str", item.a_path) for item in self.repo.index.diff(None)]
 
     @property
-    def ahead_commits(self):
-        if not self.is_valid():
+    def ahead_commits(self) -> list[Commit] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         try:
-            remote, branch_name = self.get_tracking_branch_remote()
+            tracking_branch_info = self.get_tracking_branch_remote()
+            if tracking_branch_info is None:
+                return None
+
+            remote, branch_name = tracking_branch_info
             remote_branch = f"{remote.name}/{branch_name}"
 
             return list(self.repo.iter_commits(f"{remote_branch}..{branch_name}"))
         except Exception:
             return []
 
-    def get_tracking_branch_remote(self):
-        if not self.is_valid():
+    def get_tracking_branch_remote(self) -> tuple[Remote, str] | None:
+        if self.repo is None or not self.is_valid():
             return None
 
         tracking_branch = self.tracking_branch

streamlit/hello/dataframe_demo.py

@@ -21,7 +21,7 @@ import streamlit as st
 from streamlit.hello.utils import show_code
 
 
-def data_frame_demo():
+def data_frame_demo() -> None:
     @st.cache_data
     def get_un_data() -> pd.DataFrame:
         AWS_BUCKET_URL = "https://streamlit-demo-data.s3-us-west-2.amazonaws.com"

streamlit/hello/mapping_demo.py

@@ -21,7 +21,7 @@ import streamlit as st
 from streamlit.hello.utils import show_code
 
 
-def mapping_demo():
+def mapping_demo() -> None:
     @st.cache_data
     def from_data_file(filename: str) -> pd.DataFrame:
         url = (

streamlit/hello/plotting_demo.py

@@ -20,14 +20,14 @@ import streamlit as st
 from streamlit.hello.utils import show_code
 
 
-def plotting_demo():
+def plotting_demo() -> None:
     progress_bar = st.sidebar.progress(0)
     status_text = st.sidebar.empty()
-    last_rows = np.random.randn(1, 1)
+    last_rows = np.random.randn(1, 1)  # noqa: NPY002
     chart = st.line_chart(last_rows)
 
     for i in range(1, 101):
-        new_rows = last_rows[-1, :] + np.random.randn(5, 1).cumsum(axis=0)
+        new_rows = last_rows[-1, :] + np.random.randn(5, 1).cumsum(axis=0)  # noqa: NPY002
         status_text.text(f"{i}% complete")
         chart.add_rows(new_rows)
         progress_bar.progress(i)

streamlit/hello/streamlit_app.py

@@ -20,7 +20,7 @@ dir_path = Path(__file__).parent
 
 
 # Note that this needs to be in a method so we can have an e2e playwright test.
-def run():
+def run() -> None:
     page = st.navigation(
         [
             st.Page(