fix: method not accepting None being called with None.

matplolib `Axes.set_xlabel` does not support `None` as the first parameter. https://matplotlib.org/stable/api/_as_gen/matplotlib.axes.Axes.set_xlabel.html
datamole-ai · Jul 31, 2023 · aedf276 · aedf276
1 parent e049373
commit aedf276
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/edvart/plots.py b/edvart/plots.py
@@ -158,8 +158,10 @@ def scatter_plot_2d(
     if interactive:
         fig.show()
     else:
-        ax.set_xlabel(xlabel)
-        ax.set_ylabel(ylabel)
+        if xlabel is not None:
+            ax.set_xlabel(xlabel)
+        if ylabel is not None:
+            ax.set_ylabel(ylabel)
         if not show_xticks:
             ax.set_xticks([])
         if not show_yticks: