pylint and changed comparison for saving
@@ -32,7 +32,7 @@ def _write_to_group(group, dictionary):
if isinstance(value, str):
group.attrs[key] = value
elif isinstance(value, np.ndarray):
- if value.dtype in ['<U2', '<U5']:
+ if str(value.dtype)[:2] == '<U':
group[key] = value.astype('S')
else:
group[key] = value
