un-require optional packages

Author: ATL2001

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spatial-polars"
-version = "0.2.1" # dont forget __init__.py and uv lock
+version = "0.2.2" # dont forget __init__.py and uv lock
 description = "A package that extends polars for working with geospatial data."
 readme = "README.md"
 authors = [

src/spatial_polars/_utils.py

@@ -4,10 +4,22 @@
 
 import numpy as np
 import polars as pl
-from matplotlib.colors import Colormap
-from palettable.palette import Palette
 from polars import selectors as cs
 
+try:
+    from matplotlib.colors import Colormap
+
+    HAS_MATPLOTLIB = True
+except ModuleNotFoundError:
+    HAS_MATPLOTLIB = False
+
+try:
+    from palettable.palette import Palette
+
+    HAS_PALETTABLE = True
+except ModuleNotFoundError:
+    HAS_PALETTABLE = False
+
 
 def validate_col_exists_in_df(df: pl.DataFrame, col: str) -> None:
     if col not in df.columns:
@@ -138,7 +150,9 @@ def validate_cmap_input(
         raise ValueError(msg)
 
     if cmap_type == "continuous":
-        if isinstance(cmap, (Palette, Colormap)) is False:
+        if (HAS_MATPLOTLIB and isinstance(cmap, Colormap) is False) and (
+            HAS_PALETTABLE and isinstance(cmap, Palette) is False
+        ):
             # verify continuous cmaps get a cmap or appropriate type.
             msg = (
                 f"`{cmap_type=}` is `continuous`, but {cmap=}.  Please provide a",

src/spatial_polars/spatialframe.py

@@ -16,8 +16,6 @@
 import pyproj
 import shapely
 from geoarrow.pyarrow import io as gaio
-from lonboard import PathLayer, PolygonLayer, ScatterplotLayer, viz
-from lonboard.colormap import apply_categorical_cmap, apply_continuous_cmap
 from polars import col as c
 
 from ._utils import validate_cmap_input, validate_width_and_radius_input
@@ -43,6 +41,33 @@
     "SpatialFrame",
 ]
 
+try:
+    from lonboard import PathLayer, PolygonLayer, ScatterplotLayer, viz
+    from lonboard.colormap import apply_categorical_cmap, apply_continuous_cmap
+
+    HAS_LONBOARD = True
+except ModuleNotFoundError:
+    HAS_LONBOARD = False
+
+try:
+    from scipy.spatial import KDTree
+
+    HAS_SCIPY = True
+except ModuleNotFoundError:
+    HAS_SCIPY = False
+
+
+def check_lonboard() -> None:
+    if HAS_LONBOARD is False:
+        err = "Lonboard not installed, to use this function, please install lonboard."
+        raise ModuleNotFoundError(err)
+
+
+def check_scipy() -> None:
+    if HAS_SCIPY is False:
+        err = "Scipy not installed, to use this function, please install scipy."
+        raise ModuleNotFoundError(err)
+
 
 @pl.api.register_dataframe_namespace("spatial")
 class SpatialFrame:
@@ -691,8 +716,6 @@ def centroid_knn_join(
             This method relies on scipy.spatial's KDTree to find the neighbors.
 
         """
-        from scipy.spatial import KDTree  # NOQA:PLC0415
-
         if left_on is None:
             left_on = on
         if right_on is None:
@@ -771,6 +794,7 @@ def viz(
         >>> df.spatial.viz()
 
         """  # NOQA:E501
+        check_lonboard()
         geoarrow_table = self.to_geoarrow(geometry_name)
 
         return viz(
@@ -790,6 +814,7 @@ def _make_color_array(
         *,
         normalize_cmap_col: bool,
     ) -> NDArray | None:
+        check_lonboard()
         color = None
         if cmap_col is not None:
             if cmap_type == "continuous":
@@ -976,6 +1001,7 @@ def to_scatterplotlayer(
         Implementation varies slightly from Lonboard for the setting of color and width to make it easy to use from the SpatialFrame.
 
         """  # NOQA:E501
+        check_lonboard()
         validate_cmap_input(
             self._df,
             fill_cmap_col,
@@ -1145,6 +1171,7 @@ def to_pathlayer(
         Implementation varies slightly from Lonboard for the setting of color and width to make it easy to use from the SpatialFrame.
 
         """  # NOQA:E501
+        check_lonboard()
         validate_cmap_input(
             self._df,
             cmap_col,
@@ -1329,6 +1356,7 @@ def to_polygonlayer(
         Implementation varies slightly from Lonboard for the setting of color and width to make it easy to use from the SpatialFrame.
 
         """  # NOQA:E501
+        check_lonboard()
         validate_cmap_input(
             self._df,
             fill_cmap_col,