Dont fail on bounds (#68)

* Don't fail if the bounds don't generate properly * Bump version
randyzwitch · May 26, 2022 · f4f3bde · f4f3bde
1 parent 4c5f69a
commit f4f3bde
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setuptools.setup(
     name="streamlit_folium",
-    version="0.6.11",
+    version="0.6.12",
     author="Randy Zwitch",
     author_email="[email protected]",
     description="Render Folium objects in Streamlit",

diff --git a/streamlit_folium/__init__.py b/streamlit_folium/__init__.py
@@ -146,6 +146,11 @@ def bounds_to_dict(bounds_list: List[List[float]]) -> Dict[str, Dict[str, float]
             },
         }
 
+    try:
+        bounds = fig.get_bounds()
+    except AttributeError:
+        bounds = [[None, None], [None, None]]
+
     component_value = _component_func(
         fig=leaflet,
         id=m_id,
@@ -157,7 +162,7 @@ def bounds_to_dict(bounds_list: List[List[float]]) -> Dict[str, Dict[str, float]
             "last_object_clicked": None,
             "all_drawings": None,
             "last_active_drawing": None,
-            "bounds": bounds_to_dict(fig.get_bounds()),
+            "bounds": bounds_to_dict(bounds),
             "zoom": fig.options.get("zoom") if hasattr(fig, "options") else {},
             "last_circle_radius": None,
             "last_circle_polygon": None,