streamlit-nightly 1.53.1.dev20260114__tar.gz → 1.53.1.dev20260116__tar.gz

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-nightly
-Version: 1.53.1.dev20260114
+Version: 1.53.1.dev20260116
 Summary: A faster way to build and share data apps
 Home-page: https://streamlit.io
 Author: Snowflake Inc

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/setup.py

@@ -23,7 +23,7 @@ from setuptools.command.install import install
 
 THIS_DIRECTORY = Path(__file__).parent
 
-VERSION = "1.53.1.dev20260114"  # PEP-440
+VERSION = "1.53.1.dev20260116"  # PEP-440
 
 # IMPORTANT: We should try very hard *not* to add dependencies to Streamlit.
 # And if you do add one, make the required version as general as possible:

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/__init__.py

@@ -60,7 +60,6 @@ _os.environ["MPLBACKEND"] = "Agg"
 # Must be at the top, to avoid circular dependency.
 from streamlit import logger as _logger
 from streamlit import config as _config
-from streamlit.deprecation_util import deprecate_func_name as _deprecate_func_name
 from streamlit.version import STREAMLIT_VERSION_STRING as _STREAMLIT_VERSION_STRING
 
 # Give the package a version.
@@ -116,10 +115,6 @@ from streamlit.user_info import (
     login as _login,
     logout as _logout,
 )
-from streamlit.commands.experimental_query_params import (
-    get_query_params as _get_query_params,
-    set_query_params as _set_query_params,
-)
 
 import streamlit.column_config as _column_config
 
@@ -283,27 +278,6 @@ user = _UserInfoProxy()
 # Experimental APIs
 experimental_user = _DeprecatedUserInfoProxy()
 
-_EXPERIMENTAL_QUERY_PARAMS_DEPRECATE_MSG = (
-    "Refer to our [docs page](https://docs.streamlit.io/develop/api-reference/caching-and-state/st.query_params) "
-    "for more information."
-)
-
-experimental_get_query_params = _deprecate_func_name(
-    _get_query_params,
-    "experimental_get_query_params",
-    "2024-04-11",
-    _EXPERIMENTAL_QUERY_PARAMS_DEPRECATE_MSG,
-    name_override="query_params",
-)
-experimental_set_query_params = _deprecate_func_name(
-    _set_query_params,
-    "experimental_set_query_params",
-    "2024-04-11",
-    _EXPERIMENTAL_QUERY_PARAMS_DEPRECATE_MSG,
-    name_override="query_params",
-)
-
-
 # make it possible to call streamlit.components.v1.html etc. by importing it here
 # import in the very end to avoid partially-initialized module import errors, because
 # streamlit.components.v1 also uses some streamlit imports

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/auth_util.py

@@ -19,7 +19,7 @@ import re
 from collections.abc import Callable, Mapping
 from datetime import datetime, timedelta, timezone
 from typing import TYPE_CHECKING, Any, Final, TypedDict, cast
-from urllib.parse import urlparse
+from urllib.parse import urlencode, urlparse
 
 from streamlit import config
 from streamlit.errors import StreamlitAuthError
@@ -148,6 +148,95 @@ def get_redirect_uri(auth_section: AttrDict) -> str | None:
     return redirect_uri_parsed.geturl()
 
 
+def get_validated_redirect_uri() -> str | None:
+    """Get the redirect_uri from secrets, validating it ends with /oauth2callback.
+
+    This is used for logout flows where we need a validated redirect URI
+    that matches the OAuth callback path.
+
+    Returns
+    -------
+    str | None
+        The validated redirect URI, or None if not configured or invalid.
+    """
+    auth_section = get_secrets_auth_section()
+    if not auth_section:
+        return None
+
+    redirect_uri = get_redirect_uri(auth_section)
+    if not redirect_uri:
+        return None
+
+    if not redirect_uri.endswith("/oauth2callback"):
+        _LOGGER.warning("Redirect URI does not end with /oauth2callback")
+        return None
+
+    return redirect_uri
+
+
+def get_origin_from_redirect_uri() -> str | None:
+    """Extract the origin (scheme + host) from the configured redirect_uri.
+
+    Returns
+    -------
+    str | None
+        The origin in format "scheme://host:port", or None if not configured.
+    """
+    auth_section = get_secrets_auth_section()
+    if not auth_section:
+        return None
+
+    redirect_uri = get_redirect_uri(auth_section)
+    if not redirect_uri:
+        return None
+
+    redirect_uri_parsed = urlparse(redirect_uri)
+    return f"{redirect_uri_parsed.scheme}://{redirect_uri_parsed.netloc}"
+
+
+def build_logout_url(
+    end_session_endpoint: str,
+    client_id: str,
+    post_logout_redirect_uri: str,
+    id_token: str | None = None,
+) -> str:
+    """Build an OIDC logout URL with the required parameters.
+
+    Parameters
+    ----------
+    end_session_endpoint
+        The OIDC provider's end_session_endpoint URL.
+    client_id
+        The OAuth client ID.
+    post_logout_redirect_uri
+        The URI to redirect to after logout.
+    id_token
+        Optional ID token to include as id_token_hint for the logout request.
+
+    Returns
+    -------
+    str
+        The complete logout URL with query parameters.
+    """
+    from urllib.parse import parse_qsl
+
+    logout_params: dict[str, str] = {
+        "client_id": client_id,
+        "post_logout_redirect_uri": post_logout_redirect_uri,
+    }
+
+    if id_token:
+        logout_params["id_token_hint"] = id_token
+
+    # Per OIDC spec, end_session_endpoint should be a clean URL without query params,
+    # but we handle existing params defensively for non-standard providers.
+    parsed = urlparse(end_session_endpoint)
+    existing_params = dict(parse_qsl(parsed.query))
+    merged_params = {**existing_params, **logout_params}
+    new_query = urlencode(merged_params)
+    return parsed._replace(query=new_query).geturl()
+
+
 def encode_provider_token(provider: str) -> str:
     """Returns a signed JWT token with the provider and expiration time."""
     try:

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/cli_util.py

@@ -17,7 +17,6 @@
 from __future__ import annotations
 
 import os
-import subprocess
 from typing import Any
 
 from streamlit import env_util, errors
@@ -61,6 +60,8 @@ def _open_browser_with_webbrowser(url: str) -> None:
 def _open_browser_with_command(command: str, url: str) -> None:
     cmd_line = [command, url]
     with open(os.devnull, "w", encoding="utf-8") as devnull:
+        import subprocess  # noqa: S404
+
         subprocess.Popen(cmd_line, stdout=devnull, stderr=subprocess.STDOUT)  # noqa: S603
 
 

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/commands/echo.py

@@ -26,8 +26,8 @@ from streamlit.runtime.metrics_util import gather_metrics
 if TYPE_CHECKING:
     from collections.abc import Generator, Iterable
 
-_SPACES_RE = re.compile("\\s*")
-_EMPTY_LINE_RE = re.compile("\\s*\n")
+_SPACES_RE = re.compile(r"\s*")
+_EMPTY_LINE_RE = re.compile(r"\s*\n")
 
 
 @gather_metrics("echo")

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/commands/execution_control.py

@@ -158,7 +158,7 @@ def rerun(  # type: ignore[misc]
 
     """
 
-    if scope not in ["app", "fragment"]:
+    if scope not in {"app", "fragment"}:
         raise StreamlitAPIException(
             f"'{scope}'is not a valid rerun scope. Valid scopes are 'app' and 'fragment'."
         )

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/commands/navigation.py

@@ -292,7 +292,7 @@ def navigation(
 
     """
     # Validate position parameter
-    if not isinstance(position, str) or position not in ["sidebar", "hidden", "top"]:
+    if not isinstance(position, str) or position not in {"sidebar", "hidden", "top"}:
         raise StreamlitAPIException(
             f'Invalid position "{position}". '
             'The position parameter must be one of "sidebar", "hidden", or "top".'

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/components/types/base_custom_component.py

@@ -28,8 +28,6 @@ if TYPE_CHECKING:
 class MarshallComponentException(StreamlitAPIException):
     """Class for exceptions generated during custom component marshalling."""
 
-    pass
-
 
 class BaseCustomComponent(ABC):
     """Interface for CustomComponents."""

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/components/v1/custom_component.py

@@ -40,8 +40,6 @@ if TYPE_CHECKING:
 class MarshallComponentException(StreamlitAPIException):
     """Class for exceptions generated during custom component marshalling."""
 
-    pass
-
 
 class CustomComponent(BaseCustomComponent):
     """A Custom Component declaration."""

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/components/v2/component_path_utils.py

@@ -146,7 +146,7 @@ class ComponentPathUtils:
             len(path) >= 3
             and path[0].isalpha()
             and path[1] == ":"
-            and path[2] in ("/", "\\")
+            and path[2] in {"/", "\\"}
         )
         is_unc_abs = path.startswith("\\\\")
 

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/components/v2/manifest_scanner.py

@@ -187,8 +187,13 @@ def _is_likely_streamlit_component_package(
     bool
         True if the package might contain streamlit components, False otherwise.
     """
-    # Get package metadata
-    name = dist.name.lower()
+    dist_name = dist.name
+    if not isinstance(dist_name, str) or not dist_name:
+        # We do not expect a distribution to be missing its name, be defensive
+        # and fail closed to prevent runtime issues.
+        return False
+
+    name = dist_name.lower()
     summary = dist.metadata["Summary"].lower() if "Summary" in dist.metadata else ""
 
     # Filter 1: Package name suggests streamlit component
@@ -386,7 +391,7 @@ def _validate_pyproject_for_package(
             canonical_package = packaging_utils.canonicalize_name(package_name)
 
             # Check if project name matches either the distribution name or the package name
-            return canonical_project in (canonical_dist, canonical_package)
+            return canonical_project in {canonical_dist, canonical_package}
 
         # If we can't determine ownership, be conservative and reject it
         return False

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/components/v2/presentation.py

@@ -127,7 +127,7 @@ def make_bidi_component_presenter(
                 # st.session_state[component_user_key][name] = value. Using a
                 # dict subclass ensures pretty-printing and JSON serialization
                 # behave as expected for st.write and logs.
-                class _WriteThrough(dict[str, object]):
+                class _WriteThrough(dict[str, object]):  # noqa: FURB189
                     def __init__(self, data: dict[str, object]) -> None:
                         super().__init__(data)
 

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/config.py

@@ -99,11 +99,11 @@ class ShowErrorDetailsConfigOptions(str, Enum):
 
     @staticmethod
     def is_true_variation(val: str | bool) -> bool:
-        return val in ["true", "True", True]
+        return val in {"true", "True", True}
 
     @staticmethod
     def is_false_variation(val: str | bool) -> bool:
-        return val in ["false", "False", False]
+        return val in {"false", "False", False}
 
         # Config options can be set from several places including the command-line and
         # the user's script. Legacy config options (true/false) will have type string
@@ -1077,9 +1077,11 @@ def _browser_server_port() -> int:
 
 
 _SSL_PRODUCTION_WARNING = [
-    "DO NOT USE THIS OPTION IN A PRODUCTION ENVIRONMENT. It has not gone through "
-    "security audits or performance tests. For a production environment, we "
-    "recommend performing SSL termination through a load balancer or reverse proxy."
+    (
+        "DO NOT USE THIS OPTION IN A PRODUCTION ENVIRONMENT. It has not gone through "
+        "security audits or performance tests. For a production environment, we "
+        "recommend performing SSL termination through a load balancer or reverse proxy."
+    )
 ]
 
 _create_option(
@@ -2279,6 +2281,35 @@ _create_theme_options(
     """,
 )
 
+_create_theme_options(
+    "chartDivergingColors",
+    categories=["theme"],
+    description="""
+        An array of ten colors to use for diverging chart data.
+
+        The ten colors create a diverging color scale, typically used for data
+        with a meaningful midpoint. These colors apply to Plotly, Altair, and
+        Vega-Lite charts.
+
+        Invalid color strings are skipped. If there are not exactly ten
+        valid colors specified, Streamlit uses a default set of colors.
+
+        The default colors are:
+        [
+            "#7d353b", #red100
+            "#bd4043", #red90
+            "#ff4b4b", #red70
+            "#ff8c8c", #red50
+            "#ffc7c7", #red30
+            "#a6dcff", #blue30
+            "#60b4ff", #blue50
+            "#1c83e1", #blue70
+            "#0054a3", #blue90
+            "#004280", #blue100
+        ]
+    """,
+)
+
 # Config Section: Secrets #
 
 _create_section("secrets", "Secrets configuration.")
@@ -2347,10 +2378,10 @@ def is_manually_set(option_name: str) -> bool:
         True if the option has been set by the user.
 
     """
-    return get_where_defined(option_name) not in (
+    return get_where_defined(option_name) not in {
         ConfigOption.DEFAULT_DEFINITION,
         ConfigOption.STREAMLIT_DEFINITION,
-    )
+    }
 
 
 def show_config() -> None:
@@ -2433,19 +2464,19 @@ def _is_valid_theme_section(section_path: str) -> bool:
 
     # theme.sidebar/light/dark is valid (2 parts: "theme" + section)
     if len(parts) == 2:
-        return parts[1] in [
+        return parts[1] in {
             CustomThemeCategories.SIDEBAR.value,
             CustomThemeCategories.LIGHT.value,
             CustomThemeCategories.DARK.value,
-        ]
+        }
 
     # theme.light.sidebar/theme.dark.sidebar are the only valid 3-part patterns
     if len(parts) == 3:
         # Only allow light/dark as the middle level, with sidebar as the final level
-        if parts[1] in [
+        if parts[1] in {
             CustomThemeCategories.LIGHT.value,
             CustomThemeCategories.DARK.value,
-        ]:
+        }:
             return parts[2] == CustomThemeCategories.SIDEBAR.value
         # sidebar cannot have nested sections (theme.sidebar.light/dark)
         return False
@@ -2517,11 +2548,11 @@ def _update_config_with_toml(raw_toml: str, where_defined: str) -> None:
         for name, value in section_data.items():
             option_name = f"{section_path}.{name}"
             # Only check for nested sections when we're already in a theme section
-            if section_path.startswith("theme") and name in [
+            if section_path.startswith("theme") and name in {
                 CustomThemeCategories.SIDEBAR.value,
                 CustomThemeCategories.LIGHT.value,
                 CustomThemeCategories.DARK.value,
-            ]:
+            }:
                 # Validate the theme section before processing
                 if not _is_valid_theme_section(option_name):
                     raise StreamlitInvalidThemeSectionError(

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/config_util.py

@@ -194,7 +194,7 @@ def _clean(txt: str) -> str:
 
     Preserves leading and trailing spaces, and does not modify spaces in between lines.
     """
-    return re.sub(" +", " ", txt)
+    return re.sub(r" +", " ", txt)
 
 
 def _clean_paragraphs(txt: str) -> list[str]:
@@ -741,7 +741,7 @@ def process_theme_inheritance(
     base_value = base_option.value
 
     # Check if it's a file path or URL (not just "light" or "dark")
-    if base_value in ("light", "dark"):
+    if base_value in {"light", "dark"}:
         return
 
     def _raise_invalid_nested_base() -> None:
@@ -756,7 +756,7 @@ def process_theme_inheritance(
 
         # Validate that theme.base of the referenced theme file doesn't reference another file
         theme_base = theme_file_content.get("theme", {}).get("base")
-        if theme_base and theme_base not in ("light", "dark"):
+        if theme_base and theme_base not in {"light", "dark"}:
             _raise_invalid_nested_base()
 
         # Get current theme options from main config.toml
@@ -777,10 +777,10 @@ def process_theme_inheritance(
                     opt_name.startswith("theme.")
                     and opt_name != "theme.base"
                     and opt_config.where_defined
-                    in (
+                    in {
                         "environment variable",
                         "command-line argument or environment variable",
-                    )
+                    }
                 ):
                     high_precedence_theme_options[opt_name] = {
                         "value": opt_config.value,

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/dataframe_util.py

@@ -464,7 +464,7 @@ def _is_list_of_scalars(data: Iterable[Any]) -> bool:
 
     # Overview on all value that are interpreted as scalar:
     # https://pandas.pydata.org/docs/reference/api/pandas.api.types.is_scalar.html
-    return infer_dtype(data, skipna=True) not in ["mixed", "unknown-array"]
+    return infer_dtype(data, skipna=True) not in {"mixed", "unknown-array"}
 
 
 def _iterable_to_list(
@@ -1077,10 +1077,10 @@ def is_colum_type_arrow_incompatible(column: Series[Any] | Index[Any]) -> bool:
         # https://pandas.pydata.org/docs/reference/api/pandas.api.types.infer_dtype.html
         inferred_type = infer_dtype(column, skipna=True)
 
-        if inferred_type in [
+        if inferred_type in {
             "mixed-integer",
             "complex",
-        ]:
+        }:
             return True
         if inferred_type == "mixed":
             # This includes most of the more complex/custom types (objects, dicts,
@@ -1403,11 +1403,11 @@ def convert_pandas_df_to_data_format(
         return _unify_missing_values(df).to_dict(orient="list")
     if data_format == DataFormat.COLUMN_SERIES_MAPPING:
         return df.to_dict(orient="series")
-    if data_format in [
+    if data_format in {
         DataFormat.LIST_OF_VALUES,
         DataFormat.TUPLE_OF_VALUES,
         DataFormat.SET_OF_VALUES,
-    ]:
+    }:
         df = _unify_missing_values(df)
         return_list = []
         if len(df.columns) == 1:

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/arrow.py

@@ -647,7 +647,7 @@ class ArrowMixin:
         """
         import pyarrow as pa
 
-        if on_select not in ["ignore", "rerun"] and not callable(on_select):
+        if on_select not in {"ignore", "rerun"} and not callable(on_select):
             raise StreamlitAPIException(
                 f"You have passed {on_select} to `on_select`. But only 'ignore', "
                 "'rerun', or a callable is supported."
@@ -744,13 +744,18 @@ class ArrowMixin:
         elif (
             # Hide index column if row selections are activated and the dataframe has a range index.
             # The range index usually does not add a lot of value.
-            is_selection_activated
-            and selection_mode in ["multi-row", "single-row"]
-            and has_range_index
+            is_selection_activated and has_range_index
         ):
-            update_column_config(
-                column_config_mapping, INDEX_IDENTIFIER, {"hidden": True}
+            # Normalize selection_mode to a set to check for row selection modes
+            mode_set = (
+                {selection_mode}
+                if isinstance(selection_mode, str)
+                else set(selection_mode)
             )
+            if mode_set & {"multi-row", "single-row"}:
+                update_column_config(
+                    column_config_mapping, INDEX_IDENTIFIER, {"hidden": True}
+                )
 
         marshall_column_config(proto, column_config_mapping)
 

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/deck_gl_json_chart.py

@@ -531,7 +531,7 @@ class PydeckMixin:
         key = to_key(key)
         is_selection_activated = on_select != "ignore"
 
-        if on_select not in ["ignore", "rerun"] and not callable(on_select):
+        if on_select not in {"ignore", "rerun"} and not callable(on_select):
             raise StreamlitAPIException(
                 f"You have passed {on_select} to `on_select`. "
                 "But only 'ignore', 'rerun', or a callable is supported."

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/exception.py

@@ -278,8 +278,10 @@ def _get_stack_trace_str_list(exception: BaseException) -> list[str]:
     # Format the extracted traceback and add it to the protobuf element.
     if extracted_traceback is None:
         trace_str_list = [
-            "Cannot extract the stack trace for this exception. "
-            "Try calling exception() within the `catch` block."
+            (
+                "Cannot extract the stack trace for this exception. "
+                "Try calling exception() within the `catch` block."
+            )
         ]
     else:
         internal_frames, external_frames = _split_internal_streamlit_frames(

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/form.py

@@ -413,7 +413,7 @@ class FormMixin:
             width = "stretch" if use_container_width else "content"
 
         # Checks whether the entered button type is one of the allowed options
-        if type not in ["primary", "secondary", "tertiary"]:
+        if type not in {"primary", "secondary", "tertiary"}:
             raise StreamlitAPIException(
                 'The type argument to st.form_submit_button must be "primary", "secondary", or "tertiary". \n'
                 f'The argument passed was "{type}".'

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/layouts.py

@@ -1036,7 +1036,7 @@ class LayoutsMixin:
             width = "stretch" if use_container_width else "content"
 
         # Checks whether the entered button type is one of the allowed options
-        if type not in ["primary", "secondary", "tertiary"]:
+        if type not in {"primary", "secondary", "tertiary"}:
             raise StreamlitAPIException(
                 'The type argument to st.popover must be "primary", "secondary", or "tertiary". '
                 f'\nThe argument passed was "{type}".'

{streamlit_nightly-1.53.1.dev20260114 → streamlit_nightly-1.53.1.dev20260116}/streamlit/elements/lib/built_in_chart_utils.py

@@ -136,7 +136,7 @@ def maybe_raise_stack_warning(
     stack: bool | ChartStackType | None, command: str | None, docs_link: str
 ) -> None:
     # Check that the stack parameter is valid, raise more informative error if not
-    if stack not in (None, True, False, "normalize", "center", "layered"):
+    if stack not in {None, True, False, "normalize", "center", "layered"}:
         raise StreamlitAPIException(
             f"Invalid value for stack parameter: {stack}. Stack must be one of True, "
             'False, "normalize", "center", "layered" or None. See documentation '
@@ -425,13 +425,13 @@ def _infer_vegalite_type(
     # requires Pandas 1.3.
     typ = infer_dtype(data)
 
-    if typ in [
+    if typ in {
         "floating",
         "mixed-integer-float",
         "integer",
         "mixed-integer",
         "complex",
-    ]:
+    }:
         return "quantitative"
 
     if typ == "categorical" and data.cat.ordered:
@@ -442,9 +442,9 @@ def _infer_vegalite_type(
         # Altair already extracts the correct sort order somewhere else.
         # More info about the issue here: https://github.com/streamlit/streamlit/issues/7776
         return "ordinal"
-    if typ in ["string", "bytes", "categorical", "boolean", "mixed", "unicode"]:
+    if typ in {"string", "bytes", "categorical", "boolean", "mixed", "unicode"}:
         return "nominal"
-    if typ in [
+    if typ in {
         "datetime",
         "datetime64",
         "timedelta",
@@ -452,7 +452,7 @@ def _infer_vegalite_type(
         "date",
         "time",
         "period",
-    ]:
+    }:
         return "temporal"
     # STREAMLIT MOD: I commented this out since Streamlit doesn't use warnings.warn.
     # > warnings.warn(
@@ -913,7 +913,7 @@ def _get_axis_encodings(
     _update_encoding_with_stack(stack, stack_encoding)
 
     # Handle sorting - only relevant for bar charts
-    if chart_type in (ChartType.VERTICAL_BAR, ChartType.HORIZONTAL_BAR):
+    if chart_type in {ChartType.VERTICAL_BAR, ChartType.HORIZONTAL_BAR}:
         _update_encoding_with_sort(sort_from_user, sort_encoding)
 
     return x_encoding, y_encoding