changed typecheck import

Author: timtreis

src/spatialdata_plot/pl/render.py

@@ -68,7 +68,7 @@
 def _coerce_categorical_source(cat_source: Any) -> pd.Categorical:
     """Return a pandas Categorical from known, concrete sources only."""
     if isinstance(cat_source, dd.Series):
-        if dd.api.types.is_categorical_dtype(cat_source.dtype) and getattr(cat_source.cat, "known", True) is False:
+        if pd.api.types.is_categorical_dtype(cat_source.dtype) and getattr(cat_source.cat, "known", True) is False:
             cat_source = cat_source.cat.as_known()
         cat_source = cat_source.compute()
 

tests/pl/test_render_shapes.py

@@ -802,13 +802,13 @@ def test_plot_can_annotate_shapes_with_nan_in_table_obs_categorical_datashader(
     def test_plot_can_annotate_shapes_with_nan_in_table_obs_continuous(
         self, sdata_blobs_shapes_with_nans_in_table: SpatialData
     ):
-        sdata_blobs_shapes_with_nans_in_table.pl.render_shapes("blobs_polygons", color="cola").pl.show()
+        sdata_blobs_shapes_with_nans_in_table.pl.render_shapes("blobs_polygons", color="col_a").pl.show()
 
     def test_plot_can_annotate_shapes_with_nan_in_table_obs_continuous_datashader(
         self, sdata_blobs_shapes_with_nans_in_table: SpatialData
     ):
         sdata_blobs_shapes_with_nans_in_table.pl.render_shapes(
-            "blobs_polygons", color="cola", method="datashader"
+            "blobs_polygons", color="col_a", method="datashader"
         ).pl.show()
 
     def test_plot_can_annotate_shapes_with_nan_in_table_X_continuous(