add: comments

pycroscopy · Aug 18, 2023 · e1dbcec · e1dbcec
1 parent 16d7976
commit e1dbcec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sidpy/viz/dataset_viz.py b/sidpy/viz/dataset_viz.py
@@ -1010,7 +1010,7 @@ def get_xy(self):
 class ComplexSpectralImageVisualizer(object):
     """
     ### Interactive spectrum imaging plot for Complex Data
-    ## 4D support added - Utkarsh Pratiush
+    ## 4D and complex data also works
 
     """
 
@@ -1036,8 +1036,8 @@ def __init__(self, dset, figure=None, horizontal=True, **kwargs):
         else:
             self.fig = figure
 
-        # if len(dset.shape) > 4:
-        #     raise TypeError('dataset must have four dimensions at max')
+        if len(dset.shape) > 4:
+            raise TypeError('dataset must have four dimensions at max')
         if 'complex' not in dset.dtype.name:
             raise TypeError('This visualizer is only for Complex Data, data type is {}'.format(dset.dtype))