From 7f7a28c5d991a1b7e9e6fbe6c5f3ead14160d151 Mon Sep 17 00:00:00 2001 From: Jia Yu Date: Wed, 8 May 2024 15:25:31 -0700 Subject: [PATCH] Fix --- python/sedona/maps/SedonaMapUtils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/python/sedona/maps/SedonaMapUtils.py b/python/sedona/maps/SedonaMapUtils.py index 5ab875d83d..52cc47402e 100644 --- a/python/sedona/maps/SedonaMapUtils.py +++ b/python/sedona/maps/SedonaMapUtils.py @@ -15,8 +15,6 @@ # specific language governing permissions and limitations # under the License. - -import geopandas as gpd import json from sedona.sql.types import GeometryType @@ -38,6 +36,11 @@ def __convert_to_gdf_or_pdf__(cls, df, rename=True, geometry_col=None): pandas_df = df.toPandas() if geometry_col is None: # No geometry column found even after searching schema, return Pandas Dataframe return pandas_df + try: + import geopandas as gpd + except ImportError: + msg = "GeoPandas is missing. You can install it manually or via sedona[kepler-map] or sedona[pydeck-map]." + raise ImportError(msg) from None geo_df = gpd.GeoDataFrame(pandas_df, geometry=geometry_col) if geometry_col != "geometry" and rename is True: geo_df.rename_geometry("geometry", inplace=True)