Update _base.py (#1140)

wright-group · Sep 17, 2023 · 6eff944 · 6eff944
1 parent ff51ebf
commit 6eff944
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/WrightTools/artists/_base.py b/WrightTools/artists/_base.py
@@ -130,7 +130,7 @@ def _parse_plot_args(self, *args, **kwargs):
                     raise wt_exceptions.ValueError("Cannot squeeze axis to fit channel")
             zi = data.channels[channel_index].points
             if not zi.ndim == 2:
-                raise wt_exceptions.DimensionalityError(zi.ndim, data.ndim)
+                raise wt_exceptions.DimensionalityError(2, zi.ndim)
             squeeze = tuple([0 if i else slice(None) for i in squeeze])
             if plot_type == "imshow":
                 if "aspect" not in kwargs.keys():
@@ -646,7 +646,7 @@ def plot(self, *args, **kwargs):
             zi = data.channels[channel_index].points
             xi = xa[squeeze]
             if not zi.ndim == 1:
-                raise wt_exceptions.DimensionalityError(1, data.ndim)
+                raise wt_exceptions.DimensionalityError(1, zi.ndim)
             args = [xi, zi] + args
         else:
             data = None