fixing formatting

Signed-off-by: Valerie Gleason 👌 <[email protected]>

Author: hellovolcano

src/nipanel/_streamlit_panel_initializer.py

@@ -1,10 +1,8 @@
-from streamlit.components.v1 import components
 from typing import cast
 
 import streamlit as st
 
 from nipanel._streamlit_panel_value_accessor import StreamlitPanelValueAccessor
-from nipanel._streamlit_constants import STREAMLIT_REFRESH_COMPONENT_URL
 from nipanel.panel_refresh import add_refresh_component
 
 PANEL_ACCESSOR_KEY = "StreamlitPanelValueAccessor"

src/nipanel/_streamlit_panel_value_accessor.py

@@ -42,4 +42,4 @@ def __init__(
             discovery_client=discovery_client,
             grpc_channel_pool=grpc_channel_pool,
             grpc_channel=grpc_channel,
-        )
+        )

src/nipanel/panel_refresh/__init__.py

@@ -1,14 +1,17 @@
-""" Panel refresh component for Streamlit. """
+"""Panel refresh component for Streamlit."""
+
+from typing import Any
 
 import streamlit.components.v1 as components
+
 from nipanel._streamlit_constants import STREAMLIT_REFRESH_COMPONENT_URL
 
-def add_refresh_component(panel_id):
 
+def add_refresh_component(panel_id: str) -> Any:
+    """Add a refresh component to the Streamlit app."""
     refresh_component = components.declare_component(
         "panelRefreshComponent",
         url=f"{STREAMLIT_REFRESH_COMPONENT_URL}/{panel_id}",
-        key=f"panel_refresh_{panel_id}",
     )
 
     return refresh_component