Undo typing changes in realtime

python-visualization · Jun 2, 2024 · 1f55089 · 1f55089
1 parent c5ce381
commit 1f55089
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/folium/plugins/realtime.py b/folium/plugins/realtime.py
@@ -4,11 +4,11 @@
 
 from folium.elements import JSCSSMixin
 from folium.features import GeoJson
-from folium.map import FeatureGroup
+from folium.map import Layer
 from folium.utilities import JsCode, camelize, parse_options
 
 
-class Realtime(JSCSSMixin, FeatureGroup):
+class Realtime(JSCSSMixin, Layer):
     """Put realtime data on a Leaflet map: live tracking GPS units,
     sensor data or just about anything.
 
@@ -109,7 +109,7 @@ def __init__(
         get_feature_id: Union[JsCode, str, None] = None,
         update_feature: Union[JsCode, str, None] = None,
         remove_missing: bool = False,
-        container: Optional[Union[FeatureGroup, GeoJson]] = None,
+        container: Optional[GeoJson] = None,
         **kwargs,
     ):
         super().__init__()