fix plot categorical with colors, on labels (#337)

scverse · Dec 26, 2024 · 271d4bd · 271d4bd
1 parent 4666968
commit 271d4bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/napari_spatialdata/_widgets.py b/src/napari_spatialdata/_widgets.py
@@ -218,7 +218,8 @@ def _(self, vec: pd.Series, **kwargs: Any) -> dict[str, Any]:
                 color_dict = dict(zip(vec.cat.categories, colors, strict=False))
                 color_dict.update({np.nan: "#808080ff"})
             else:
-                color_dict = self.model.adata.uns[vec_color_name]
+                colors = self.model.adata.uns[vec_color_name]
+                color_dict = dict(zip(vec.cat.categories, colors.tolist(), strict=True))
         else:
             df = layer.metadata["_columns_df"]
             if vec_color_name not in df.columns: