Remove use of hasattr

nexpy · Feb 22, 2022 · 1d2686b · 1d2686b
1 parent edfd4cf
commit 1d2686b
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