streamlit 1.47.0__py3-none-any.whl → 1.48.0__py3-none-any.whl

streamlit/cli_util.py

@@ -103,4 +103,4 @@ def open_browser(url: str) -> None:
 
     import platform
 
-    raise errors.Error(f'Cannot open browser in platform "{platform.system()}"')
+    raise errors.Error(f'Cannot open browser in platform "{platform.system()}"')  # ty: ignore[unresolved-attribute]

streamlit/commands/echo.py

@@ -19,7 +19,7 @@ import contextlib
 import re
 import textwrap
 import traceback
-from typing import TYPE_CHECKING, Any, Literal
+from typing import TYPE_CHECKING, Any, Literal, cast
 
 from streamlit.runtime.metrics_util import gather_metrics
 
@@ -81,7 +81,7 @@ def echo(
             for child in ast.iter_child_nodes(node):
                 # If child doesn't have "lineno", it is not something we could display
                 if hasattr(child, "lineno"):
-                    line_to_node_map[child.lineno] = child
+                    line_to_node_map[cast("int", child.lineno)] = child
                     collect_body_statements(child)
 
         collect_body_statements(root_node)

streamlit/commands/execution_control.py

@@ -103,7 +103,7 @@ def _new_fragment_id_queue(
 def rerun(  # type: ignore[misc]
     *,  # The scope argument can only be passed via keyword.
     scope: Literal["app", "fragment"] = "app",
-) -> NoReturn:
+) -> NoReturn:  # ty: ignore[invalid-return-type]
     """Rerun the script immediately.
 
     When ``st.rerun()`` is called, Streamlit halts the current script run and

streamlit/commands/page_config.py

@@ -194,21 +194,19 @@ def set_page_config(
 
     menu_items: dict
         Configure the menu that appears on the top-right side of this app.
-        The keys in this dict denote the menu item you'd like to configure:
-
-        - "Get help": str or None
-            The URL this menu item should point to.
-            If None, hides this menu item.
-        - "Report a Bug": str or None
-            The URL this menu item should point to.
-            If None, hides this menu item.
-        - "About": str or None
-            A markdown string to show in the About dialog.
-            If None, only shows Streamlit's default About text.
-
-        The URL may also refer to an email address e.g. ``mailto:john@example.com``.
-        To remove an item that was specified in a previous call to
-        ``st.set_page_config``, set its value to ``None`` in the dictionary.
+        The keys in this dict denote the menu item to configure. The following
+        keys can have string or ``None`` values:
+
+        - "Get help": The URL this menu item should point to.
+        - "Report a Bug": The URL this menu item should point to.
+        - "About": A markdown string to show in the About dialog.
+
+        A URL may also refer to an email address e.g. ``mailto:john@example.com``.
+
+        If you do not include a key, its menu item will be hidden (unless it
+        was set by a previous call to ``st.set_page_config``). To remove an
+        item that was specified in a previous call to ``st.set_page_config``,
+        set its value to ``None`` in the dictionary.
 
     Example
     -------
@@ -301,7 +299,9 @@ def set_menu_items_proto(
 
     if ABOUT_KEY in lowercase_menu_items:
         if lowercase_menu_items[ABOUT_KEY] is not None:
-            menu_items_proto.about_section_md = dedent(lowercase_menu_items[ABOUT_KEY])
+            menu_items_proto.about_section_md = dedent(
+                lowercase_menu_items[ABOUT_KEY] or ""
+            )
         else:
             # For multiple calls to set_page_config, clears previously set about markdown
             menu_items_proto.clear_about_md = True

streamlit/components/v1/component_arrow.py

@@ -72,8 +72,8 @@ def _marshall_index(proto: ArrowTableProto, index: Index) -> None:
     """
     import pandas as pd
 
-    index = map(_maybe_tuple_to_list, index.values)
-    index_df = pd.DataFrame(index)
+    index_values = map(_maybe_tuple_to_list, index.values)
+    index_df = pd.DataFrame(index_values)
     proto.index = dataframe_util.convert_pandas_df_to_arrow_bytes(index_df)
 
 
@@ -92,8 +92,8 @@ def _marshall_columns(proto: ArrowTableProto, columns: Series) -> None:
     """
     import pandas as pd
 
-    columns = map(_maybe_tuple_to_list, columns.values)
-    columns_df = pd.DataFrame(columns)
+    values = map(_maybe_tuple_to_list, columns.values)
+    columns_df = pd.DataFrame(values)
     proto.columns = dataframe_util.convert_pandas_df_to_arrow_bytes(columns_df)
 
 

streamlit/config.py

@@ -36,8 +36,8 @@ from streamlit.errors import StreamlitAPIException
 # Descriptions of each of the possible config sections.
 # (We use OrderedDict to make the order in which sections are declared in this
 # file be the same order as the sections appear with `streamlit config show`)
-_section_descriptions: dict[str, str] = OrderedDict(
-    _test="Special test section just used for unit tests."
+_section_descriptions: OrderedDict[str, str] = OrderedDict(  # ty: ignore
+    _test="Special test section just used for unit tests."  # ty: ignore
 )
 
 # Ensures that we don't try to get or set config options when config.toml files
@@ -925,6 +925,24 @@ _create_option(
     type_=bool,
 )
 
+_create_option(
+    "server.websocketPingInterval",
+    description="""
+        The interval (in seconds) at which the server pings the client to keep
+        the websocket connection alive.
+
+        The default value should work for most deployments. However, if you're
+        experiencing frequent disconnections in certain proxy setups (e.g.,
+        "Connection error" messages), you may want to try adjusting this value.
+
+        Note: When you set this option, Streamlit automatically sets the ping
+        timeout to match this interval. For Tornado >=6.5, a value less than 30
+        may cause connection issues.
+    """,
+    default_val=None,
+    type_=int,
+)
+
 _create_option(
     "server.enableStaticServing",
     description="""
@@ -1364,13 +1382,13 @@ _create_theme_options(
 
 _create_theme_options(
     "codeFontWeight",
-    categories=["theme"],
+    categories=["theme", CustomThemeCategories.SIDEBAR],
     description="""
         The font weight for code blocks and code text.
 
         This applies to font in inline code, code blocks, `st.json`, and
         `st.help`. This is an integer multiple of 100. Values can be between
-        100 and 900, inclusive.
+        100 and 600, inclusive.
 
         If this isn't set, the code font weight will be 400 (normal weight).
     """,
@@ -1884,7 +1902,7 @@ def get_config_options(
         # Short-circuit if config files were parsed while we were waiting on
         # the lock.
         if _config_options and not force_reparse:
-            return _config_options
+            return _config_options  # ty: ignore[invalid-return-type]
 
         old_options = _config_options
         _config_options = copy.deepcopy(_config_options_template)

streamlit/connections/base_connection.py

@@ -109,11 +109,11 @@ class BaseConnection(ABC, Generic[RawConnectionT]):
         are implementing their class' ``_connect`` method. User scripts should, for the
         most part, have no reason to use this property.
         """
-        connections_section = None
+        connections_section: AttrDict | None = None
         if secrets_singleton.load_if_toml_exists():
             connections_section = secrets_singleton.get("connections")
 
-        if type(connections_section) is not AttrDict:
+        if connections_section is None or type(connections_section) is not AttrDict:
             return AttrDict({})
 
         return connections_section.get(self._connection_name, AttrDict({}))

streamlit/connections/snowflake_connection.py

@@ -68,10 +68,10 @@ class SnowflakeConnection(BaseConnection["InternalSnowflakeConnection"]):
     .. |snowflake.connector.connect()| replace:: ``snowflake.connector.connect()``
     .. _snowflake.connector.connect(): https://docs.snowflake.com/en/developer-guide/python-connector/python-connector-api#label-snowflake-connector-methods-connect
 
-    .. Tip::
+    .. Important::
         `snowflake-snowpark-python <https://pypi.org/project/snowflake-snowpark-python/>`_
         must be installed in your environment to use this connection. You can
-        install Snowflake extras along with Streamlit:
+        install it as an extra with Streamlit:
 
         >>> pip install streamlit[snowflake]
 

streamlit/connections/sql_connection.py

@@ -178,6 +178,7 @@ class SQLConnection(BaseConnection["Engine"]):
 
     def _connect(self, autocommit: bool = False, **kwargs: Any) -> Engine:
         import sqlalchemy
+        from sqlalchemy.engine import URL, make_url
 
         kwargs = deepcopy(kwargs)
         conn_param_kwargs = extract_from_dict(_ALL_CONNECTION_PARAMS, kwargs)
@@ -190,7 +191,7 @@ class SQLConnection(BaseConnection["Engine"]):
             )
 
         if "url" in conn_params:
-            url = sqlalchemy.engine.make_url(conn_params["url"])
+            url = make_url(conn_params["url"])
         else:
             for p in _REQUIRED_CONNECTION_PARAMS:
                 if p not in conn_params:
@@ -200,7 +201,7 @@ class SQLConnection(BaseConnection["Engine"]):
                 f"+{conn_params['driver']}" if "driver" in conn_params else ""
             )
 
-            url = sqlalchemy.engine.URL.create(
+            url = URL.create(
                 drivername=drivername,
                 username=conn_params["username"],
                 password=conn_params.get("password"),
@@ -217,7 +218,7 @@ class SQLConnection(BaseConnection["Engine"]):
 
         if autocommit:
             return cast("Engine", eng.execution_options(isolation_level="AUTOCOMMIT"))
-        return cast("Engine", eng)
+        return cast("Engine", eng)  # ty: ignore[redundant-cast]
 
     def query(
         self,

streamlit/dataframe_util.py

@@ -1076,7 +1076,7 @@ def is_colum_type_arrow_incompatible(column: Series[Any] | Index) -> bool:
                 return True
 
             # Get the first value to check if it is a supported list-like type.
-            first_value = column.iloc[0]
+            first_value = cast("DataFrameGenericAlias[Any]", column).iloc[0]
 
             if (  # noqa: SIM103
                 not is_list_like(first_value)

streamlit/deprecation_util.py

@@ -27,14 +27,25 @@ TFunc = TypeVar("TFunc", bound=Callable[..., Any])
 TObj = TypeVar("TObj", bound=object)
 
 
-def _should_show_deprecation_warning_in_browser() -> bool:
+def _error_details_in_browser_enabled() -> bool:
     """True if we should print deprecation warnings to the browser."""
     return bool(config.get_option("client.showErrorDetails"))
 
 
-def show_deprecation_warning(message: str) -> None:
-    """Show a deprecation warning message."""
-    if _should_show_deprecation_warning_in_browser():
+def show_deprecation_warning(message: str, show_in_browser: bool = True) -> None:
+    """Show a deprecation warning message.
+
+    Parameters
+    ----------
+    message : str
+        The deprecation warning message.
+    show_in_browser : bool, default=True
+        Whether to show the deprecation warning in the browser. When this is True,
+        we will show the deprecation warning in the browser unless the user has
+        disabled error details in the browser by setting the `client.showErrorDetails`
+        config option to "none".
+    """
+    if _error_details_in_browser_enabled() and show_in_browser:
         streamlit.warning(message)
 
     # We always log deprecation warnings
@@ -97,7 +108,11 @@ def deprecate_func_name(
         result = func(*args, **kwargs)
         show_deprecation_warning(
             make_deprecated_name_warning(
-                old_name, name_override or func.__name__, removal_date, extra_message
+                old_name,
+                name_override
+                or (str(func.__name__) if hasattr(func, "__name__") else "unknown"),
+                removal_date,
+                extra_message,
             )
         )
         return result

streamlit/elements/arrow.py

@@ -38,6 +38,9 @@ from streamlit.elements.lib.column_config_utils import (
     update_column_config,
 )
 from streamlit.elements.lib.form_utils import current_form_id
+from streamlit.elements.lib.layout_utils import (
+    LayoutConfig,
+)
 from streamlit.elements.lib.pandas_styler_utils import marshall_styler
 from streamlit.elements.lib.policies import check_widget_policies
 from streamlit.elements.lib.utils import Key, compute_and_register_element_id, to_key
@@ -104,17 +107,16 @@ class DataframeSelectionState(TypedDict, total=False):
     Try selecting some rows. To select multiple columns, hold ``Ctrl`` while
     selecting columns. Hold ``Shift`` to select a range of columns.
 
-    >>> import streamlit as st
     >>> import pandas as pd
-    >>> import numpy as np
+    >>> import streamlit as st
+    >>> from numpy.random import default_rng as rng
     >>>
-    >>> if "df" not in st.session_state:
-    >>>     st.session_state.df = pd.DataFrame(
-    ...         np.random.randn(12, 5), columns=["a", "b", "c", "d", "e"]
-    ...     )
+    >>> df = pd.DataFrame(
+    ...     rng(0).standard_normal((12, 5)), columns=["a", "b", "c", "d", "e"]
+    ... )
     >>>
     >>> event = st.dataframe(
-    ...     st.session_state.df,
+    ...     df,
     ...     key="data",
     ...     on_select="rerun",
     ...     selection_mode=["multi-row", "multi-column"],
@@ -438,13 +440,15 @@ class ArrowMixin:
         --------
         **Example 1: Display a dataframe**
 
-        >>> import streamlit as st
         >>> import pandas as pd
-        >>> import numpy as np
+        >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
-        >>> df = pd.DataFrame(np.random.randn(50, 20), columns=("col %d" % i for i in range(20)))
+        >>> df = pd.DataFrame(
+        ...     rng(0).standard_normal((50, 20)), columns=("col %d" % i for i in range(20))
+        ... )
         >>>
-        >>> st.dataframe(df)  # Same as st.write(df)
+        >>> st.dataframe(df)
 
         .. output::
            https://doc-dataframe.streamlit.app/
@@ -455,11 +459,13 @@ class ArrowMixin:
         You can also pass a Pandas Styler object to change the style of
         the rendered DataFrame:
 
-        >>> import streamlit as st
         >>> import pandas as pd
-        >>> import numpy as np
+        >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
-        >>> df = pd.DataFrame(np.random.randn(10, 20), columns=("col %d" % i for i in range(20)))
+        >>> df = pd.DataFrame(
+        ...     rng(0).standard_normal((10, 20)), columns=("col %d" % i for i in range(20))
+        ... )
         >>>
         >>> st.dataframe(df.style.highlight_max(axis=0))
 
@@ -471,34 +477,39 @@ class ArrowMixin:
 
         You can customize a dataframe via ``column_config``, ``hide_index``, or ``column_order``.
 
-        >>> import random
         >>> import pandas as pd
         >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
         >>> df = pd.DataFrame(
-        >>>     {
-        >>>         "name": ["Roadmap", "Extras", "Issues"],
-        >>>         "url": ["https://roadmap.streamlit.app", "https://extras.streamlit.app", "https://issues.streamlit.app"],
-        >>>         "stars": [random.randint(0, 1000) for _ in range(3)],
-        >>>         "views_history": [[random.randint(0, 5000) for _ in range(30)] for _ in range(3)],
-        >>>     }
-        >>> )
+        ...     {
+        ...         "name": ["Roadmap", "Extras", "Issues"],
+        ...         "url": [
+        ...             "https://roadmap.streamlit.app",
+        ...             "https://extras.streamlit.app",
+        ...             "https://issues.streamlit.app",
+        ...         ],
+        ...         "stars": rng(0).integers(0, 1000, size=3),
+        ...         "views_history": rng(0).integers(0, 5000, size=(3, 30)).tolist(),
+        ...     }
+        ... )
+        >>>
         >>> st.dataframe(
-        >>>     df,
-        >>>     column_config={
-        >>>         "name": "App name",
-        >>>         "stars": st.column_config.NumberColumn(
-        >>>             "Github Stars",
-        >>>             help="Number of stars on GitHub",
-        >>>             format="%d ⭐",
-        >>>         ),
-        >>>         "url": st.column_config.LinkColumn("App URL"),
-        >>>         "views_history": st.column_config.LineChartColumn(
-        >>>             "Views (past 30 days)", y_min=0, y_max=5000
-        >>>         ),
-        >>>     },
-        >>>     hide_index=True,
-        >>> )
+        ...     df,
+        ...     column_config={
+        ...         "name": "App name",
+        ...         "stars": st.column_config.NumberColumn(
+        ...             "Github Stars",
+        ...             help="Number of stars on GitHub",
+        ...             format="%d ⭐",
+        ...         ),
+        ...         "url": st.column_config.LinkColumn("App URL"),
+        ...         "views_history": st.column_config.LineChartColumn(
+        ...             "Views (past 30 days)", y_min=0, y_max=5000
+        ...         ),
+        ...     },
+        ...     hide_index=True,
+        ... )
 
         .. output::
            https://doc-dataframe-config.streamlit.app/
@@ -508,22 +519,28 @@ class ArrowMixin:
 
         You can use column configuration to format your index.
 
-        >>> import streamlit as st
+        >>> from datetime import datetime, date
+        >>> import numpy as np
         >>> import pandas as pd
-        >>> from datetime import date
+        >>> import streamlit as st
         >>>
-        >>> df = pd.DataFrame(
-        >>>     {
-        >>>         "Date": [date(2024, 1, 1), date(2024, 2, 1), date(2024, 3, 1)],
-        >>>         "Total": [13429, 23564, 23452],
-        >>>     }
-        >>> )
-        >>> df.set_index("Date", inplace=True)
+        >>> @st.cache_data
+        >>> def load_data():
+        >>>     year = datetime.now().year
+        >>>     df = pd.DataFrame(
+        ...         {
+        ...             "Date": [date(year, month, 1) for month in range(1, 4)],
+        ...             "Total": np.random.randint(1000, 5000, size=3),
+        ...         }
+        ...     )
+        >>>     df.set_index("Date", inplace=True)
+        >>>     return df
         >>>
+        >>> df = load_data()
         >>> config = {
-        >>>     "_index": st.column_config.DateColumn("Month", format="MMM YYYY"),
-        >>>     "Total": st.column_config.NumberColumn("Total ($)"),
-        >>> }
+        ...     "_index": st.column_config.DateColumn("Month", format="MMM YYYY"),
+        ...     "Total": st.column_config.NumberColumn("Total ($)"),
+        ... }
         >>>
         >>> st.dataframe(df, column_config=config)
 
@@ -677,12 +694,13 @@ class ArrowMixin:
         --------
         **Example 1: Display a simple dataframe as a static table**
 
-        >>> import streamlit as st
         >>> import pandas as pd
-        >>> import numpy as np
+        >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
         >>> df = pd.DataFrame(
-        ...     np.random.randn(10, 5), columns=("col %d" % i for i in range(5))
+        ...     rng(0).standard_normal(size=(10, 5)),
+        ...     columns=("col %d" % i for i in range(5)),
         ... )
         >>>
         >>> st.table(df)
@@ -693,19 +711,22 @@ class ArrowMixin:
 
         **Example 2: Display a table of Markdown strings**
 
-        >>> import streamlit as st
         >>> import pandas as pd
+        >>> import streamlit as st
         >>>
         >>> df = pd.DataFrame(
         ...     {
         ...         "Command": ["**st.table**", "*st.dataframe*"],
         ...         "Type": ["`static`", "`interactive`"],
         ...         "Docs": [
-        ...             "[:rainbow[docs]](https://docs.streamlit.io/develop/api-reference/data/st.dataframe)",
-        ...             "[:book:](https://docs.streamlit.io/develop/api-reference/data/st.table)",
+        ...             "[:rainbow[docs]](https://docs.streamlit.io"
+        ...             "/develop/api-reference/data/st.dataframe)",
+        ...             "[:open_book:](https://docs.streamlit.io"
+        ...             "/develop/api-reference/data/st.table)",
         ...         ],
         ...     }
         ... )
+        >>>
         >>> st.table(df)
 
         .. output::
@@ -728,9 +749,16 @@ class ArrowMixin:
         delta_path = self.dg._get_delta_path_str()
         default_uuid = str(hash(delta_path))
 
+        # Tables dimensions are not configurable, this ensures that
+        # styles are applied correctly on the element container in the frontend.
+        layout_config = LayoutConfig(
+            width="stretch",
+            height="content",
+        )
+
         proto = ArrowProto()
         marshall(proto, data, default_uuid)
-        return self.dg._enqueue("arrow_table", proto)
+        return self.dg._enqueue("arrow_table", proto, layout_config=layout_config)
 
     @gather_metrics("add_rows")
     def add_rows(self, data: Data = None, **kwargs: Any) -> DeltaGenerator | None:
@@ -747,32 +775,30 @@ class ArrowMixin:
 
         Example
         -------
-        >>> import streamlit as st
+        >>> import time
         >>> import pandas as pd
-        >>> import numpy as np
+        >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
         >>> df1 = pd.DataFrame(
-        ...     np.random.randn(50, 20), columns=("col %d" % i for i in range(20))
-        ... )
-        >>>
-        >>> my_table = st.table(df1)
+        >>>     rng(0).standard_normal(size=(50, 20)), columns=("col %d" % i for i in range(20))
+        >>> )
         >>>
         >>> df2 = pd.DataFrame(
-        ...     np.random.randn(50, 20), columns=("col %d" % i for i in range(20))
-        ... )
+        >>>     rng(1).standard_normal(size=(50, 20)), columns=("col %d" % i for i in range(20))
+        >>> )
         >>>
+        >>> my_table = st.table(df1)
+        >>> time.sleep(1)
         >>> my_table.add_rows(df2)
-        >>> # Now the table shown in the Streamlit app contains the data for
-        >>> # df1 followed by the data for df2.
 
         You can do the same thing with plots. For example, if you want to add
         more data to a line chart:
 
         >>> # Assuming df1 and df2 from the example above still exist...
         >>> my_chart = st.line_chart(df1)
+        >>> time.sleep(1)
         >>> my_chart.add_rows(df2)
-        >>> # Now the chart shown in the Streamlit app contains the data for
-        >>> # df1 followed by the data for df2.
 
         And for plots whose datasets are named, you can pass the data with a
         keyword argument where the key is the name:
@@ -838,30 +864,30 @@ def _arrow_add_rows(
 
     Example
     -------
-    >>> import streamlit as st
+    >>> import time
     >>> import pandas as pd
-    >>> import numpy as np
+    >>> import streamlit as st
+    >>> from numpy.random import default_rng as rng
     >>>
     >>> df1 = pd.DataFrame(
-    ...     np.random.randn(50, 20), columns=("col %d" % i for i in range(20))
-    ... )
-    >>> my_table = st.table(df1)
+    >>>     rng(0).standard_normal(size=(50, 20)), columns=("col %d" % i for i in range(20))
+    >>> )
     >>>
     >>> df2 = pd.DataFrame(
-    ...     np.random.randn(50, 20), columns=("col %d" % i for i in range(20))
-    ... )
+    >>>     rng(1).standard_normal(size=(50, 20)), columns=("col %d" % i for i in range(20))
+    >>> )
+    >>>
+    >>> my_table = st.table(df1)
+    >>> time.sleep(1)
     >>> my_table.add_rows(df2)
-    >>> # Now the table shown in the Streamlit app contains the data for
-    >>> # df1 followed by the data for df2.
 
     You can do the same thing with plots. For example, if you want to add
     more data to a line chart:
 
     >>> # Assuming df1 and df2 from the example above still exist...
     >>> my_chart = st.line_chart(df1)
+    >>> time.sleep(1)
     >>> my_chart.add_rows(df2)
-    >>> # Now the chart shown in the Streamlit app contains the data for
-    >>> # df1 followed by the data for df2.
 
     And for plots whose datasets are named, you can pass the data with a
     keyword argument where the key is the name:

streamlit/elements/deck_gl_json_chart.py

@@ -410,13 +410,13 @@ class PydeckMixin:
         Here's a chart using a HexagonLayer and a ScatterplotLayer. It uses either the
         light or dark map style, based on which Streamlit theme is currently active:
 
-        >>> import streamlit as st
         >>> import pandas as pd
-        >>> import numpy as np
         >>> import pydeck as pdk
+        >>> import streamlit as st
+        >>> from numpy.random import default_rng as rng
         >>>
-        >>> chart_data = pd.DataFrame(
-        ...     np.random.randn(1000, 2) / [50, 50] + [37.76, -122.4],
+        >>> df = pd.DataFrame(
+        ...     rng(0).standard_normal((1000, 2)) / [50, 50] + [37.76, -122.4],
         ...     columns=["lat", "lon"],
         ... )
         >>>
@@ -432,7 +432,7 @@ class PydeckMixin:
         ...         layers=[
         ...             pdk.Layer(
         ...                 "HexagonLayer",
-        ...                 data=chart_data,
+        ...                 data=df,
         ...                 get_position="[lon, lat]",
         ...                 radius=200,
         ...                 elevation_scale=4,
@@ -442,7 +442,7 @@ class PydeckMixin:
         ...             ),
         ...             pdk.Layer(
         ...                 "ScatterplotLayer",
-        ...                 data=chart_data,
+        ...                 data=df,
         ...                 get_position="[lon, lat]",
         ...                 get_color="[200, 30, 0, 160]",
         ...                 get_radius=200,