fix

Author: LucaMarconato

src/spatialdata_plot/pl/basic.py

@@ -951,8 +951,9 @@ def show(
                     if wanted_labels_on_this_cs:
                         if (table := params_copy.table_name) is not None:
                             keys = [params_copy.color] if not isinstance(params_copy.color, list) else params_copy.color
+                            assert len(keys) == 1
                             colors = sc.get.obs_df(sdata[table], keys)
-                            if isinstance(colors.dtype, pd.CategoricalDtype):
+                            if isinstance(colors[keys[0]].dtype, pd.CategoricalDtype):
                                 _maybe_set_colors(
                                     source=sdata[table],
                                     target=sdata[table],

src/spatialdata_plot/pl/render.py

@@ -454,9 +454,10 @@ def _render_points(
 
     if col_for_color is not None:
         keys = [col_for_color] if not isinstance(col_for_color, list) else col_for_color
+        assert len(keys) == 1
         cols = sc.get.obs_df(adata, keys)
         # maybe set color based on type
-        if isinstance(cols.dtype, pd.CategoricalDtype):
+        if isinstance(cols[keys[0]].dtype, pd.CategoricalDtype):
             _maybe_set_colors(
                 source=adata,
                 target=adata,