use correct imports

Author: karangattu

tests/playwright/shiny/components/chat/shiny_output/app.py

@@ -1,7 +1,7 @@
 import ipyleaflet as ipyl  # pyright: ignore[reportMissingTypeStubs]
 import pandas as pd
 import plotly.express as px  # pyright: ignore[reportMissingTypeStubs]
-from plotly.graph_objs import Figure
+from plotly.graph_objs._figure import Figure
 from shinywidgets import render_plotly, render_widget
 
 from shiny import reactive, render

tests/playwright/shiny/components/data_frame/edit/app.py

@@ -17,6 +17,7 @@
 import great_tables as gt
 import palmerpenguins  # pyright: ignore[reportMissingTypeStubs]
 import polars as pl
+from plotly.graph_objs._figure import Figure
 
 from shiny import App, Inputs, Outputs, Session, module, reactive, render, req, ui
 from shiny.render import CellPatch
@@ -264,7 +265,7 @@ def df_styles_fn(data: pd.DataFrame) -> list[StyleInfo]:
     # from shiny import reactive
 
     @render_widget
-    def country_detail_pop():
+    def country_detail_pop() -> Figure:
         import plotly.express as px
 
         return px.line(