Backport PR #2682: Fix #2681

scverse · Oct 16, 2023 · 503876d · 503876d
1 parent a20bea6
commit 503876d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scanpy/plotting/_tools/scatterplots.py b/scanpy/plotting/_tools/scatterplots.py
@@ -886,8 +886,10 @@ def pca(
             # edit axis labels in returned figure
             fig = embedding(adata, 'pca', return_fig=return_fig, **kwargs)
             for ax in fig.axes:
-                ax.set_xlabel(label_dict[ax.xaxis.get_label().get_text()])
-                ax.set_ylabel(label_dict[ax.yaxis.get_label().get_text()])
+                if xlabel := label_dict.get(ax.xaxis.get_label().get_text()):
+                    ax.set_xlabel(xlabel)
+                if ylabel := label_dict.get(ax.yaxis.get_label().get_text()):
+                    ax.set_ylabel(ylabel)
             return fig
 
         else: