Refactor TypeBounds

python-visualization · Jan 1, 2023 · fa7e3c9 · fa7e3c9
1 parent 79e0e7c
commit fa7e3c9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/folium/folium.py b/folium/folium.py
@@ -14,6 +14,7 @@
 from folium.map import FitBounds, Layer
 from folium.raster_layers import TileLayer
 from folium.utilities import (
+    TypeBounds,
     TypeJsonValue,
     _parse_size,
     parse_options,
@@ -419,7 +420,7 @@ def show_in_browser(self) -> None:
 
     def fit_bounds(
         self,
-        bounds: Sequence[Sequence[float]],
+        bounds: TypeBounds,
         padding_top_left: Optional[Sequence[float]] = None,
         padding_bottom_right: Optional[Sequence[float]] = None,
         padding: Optional[Sequence[float]] = None,

diff --git a/folium/map.py b/folium/map.py
@@ -10,6 +10,7 @@
 from jinja2 import Template
 
 from folium.utilities import (
+    TypeBounds,
     TypeJsonValue,
     camelize,
     escape_backticks,
@@ -592,7 +593,7 @@ class FitBounds(MacroElement):
 
     def __init__(
         self,
-        bounds: Sequence[Sequence[float]],
+        bounds: TypeBounds,
         padding_top_left: Optional[Sequence[float]] = None,
         padding_bottom_right: Optional[Sequence[float]] = None,
         padding: Optional[Sequence[float]] = None,

diff --git a/folium/utilities.py b/folium/utilities.py
@@ -50,6 +50,8 @@
 
 TypePathOptions = Union[bool, str, float, None]
 
+TypeBounds = Sequence[Sequence[float]]
+
 
 _VALID_URLS = set(uses_relative + uses_netloc + uses_params)
 _VALID_URLS.discard("")