Update _base.py (#1115)

wright-group · Feb 27, 2023 · 07a17ec · 07a17ec
1 parent 4e36b68
commit 07a17ec
Showing 1 changed file with 1 addition and 1 deletion.
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(ndim, data.ndim)
+                raise wt_exceptions.DimensionalityError(zi.ndim, data.ndim)
             squeeze = tuple([0 if i else slice(None) for i in squeeze])
             if plot_type == "imshow":
                 if "aspect" not in kwargs.keys():