Merge pull request #171 from rayosborn/remove-hasattr

Improve attribute checks
nexpy · Mar 14, 2022 · f0406c7 · f0406c7
2 parents edfd4cf + 1d2686b
commit f0406c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nexusformat/nexus/plot.py b/src/nexusformat/nexus/plot.py
@@ -194,8 +194,8 @@ def plot(self, data_group, fmt=None, xmin=None, xmax=None,
                     fmt = kwargs.pop('marker')
                 else:
                     fmt = 'o'
-                if hasattr(signal, 'units'):
-                    if not errors and signal.units == 'counts':
+                if 'units' in signal.attrs:
+                    if not errors and signal.attrs['units'] == 'counts':
                         errors = NXfield(np.sqrt(data))
                 if errors:
                     ebars = errors.nxdata