Make sure filetype gets set correctly for data libraries

This commit is contained in:
Paul Romano 2018-11-02 07:57:55 -05:00
parent 05e66155a2
commit dd9463c288

View file

@ -57,7 +57,7 @@ class DataLibrary(EqualityMixin):
filename = str(filename)
with h5py.File(filename, 'r') as h5file:
filetype = h5file.attrs['filetype'].decode().lstrip('data_')
filetype = h5file.attrs['filetype'].decode()[5:]
materials = list(h5file)
library = {'path': filename, 'type': filetype, 'materials': materials}