Temporary fix for nan in pl.boundaries

CSOgroup · May 8, 2024 · 852ffef · 852ffef
1 parent 4e11f5b
commit 852ffef
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/cellcharter/pl/_shape.py b/src/cellcharter/pl/_shape.py
@@ -140,13 +140,11 @@ def boundaries(
     ax = plt.gca()
     if show_cells:
         # TODO: remove after spatialdata-plot issue  #256 is fixed
-        try:
-            # TODO: spatialdata-plot doesn't support legend_loc=False to make the legend disappear.
-            sdata.pl.render_shapes(elements="cells", color=component_key).pl.show(ax=ax, legend_loc=None)
-        except TypeError:
-            sdata.tables["table"].obs[component_key] = sdata.tables["table"].obs[component_key].cat.add_categories([-1])
-            sdata.tables["table"].obs[component_key] = sdata.tables["table"].obs[component_key].fillna(-1)
-            sdata.pl.render_shapes(elements="cells", color=component_key).pl.show(ax=ax, legend_loc=None)
+        sdata.tables["table"].obs[component_key] = sdata.tables["table"].obs[component_key].cat.add_categories([-1])
+        sdata.tables["table"].obs[component_key] = sdata.tables["table"].obs[component_key].fillna(-1)
+
+        # TODO: spatialdata-plot doesn't support legend_loc=False to make the legend disappear.
+        sdata.pl.render_shapes(elements="cells", color=component_key).pl.show(ax=ax, legend_loc=None)
 
     sdata.pl.render_shapes(
         elements="clusters",