diff --git a/opacplot2/opg_tops.py b/opacplot2/opg_tops.py index 8b6f825..0f095de 100644 --- a/opacplot2/opg_tops.py +++ b/opacplot2/opg_tops.py @@ -104,9 +104,10 @@ def __init__(self, filename, ep_max='auto', handle_large='next_group'): } ) dats.reshape(self.Nm) - self.Xnum = dats['Xnum'] + # Made some tweaks to ensure Xnum and Znum are iterable + self.Xnum = dats['Xnum'] if dats['Xnum'].size >1 else [dats['Xnum'].tolist()] self.Massfrac = dats['Massfrac'] - self.Znum = dats['Znum'] + self.Znum = dats['Znum'] if dats['Znum'].size >1 else [dats['Znum'].tolist()] self.Zsymb = dats['Zsymb'] self.MatID = dats['MatID'] self.Zmax = np.average(self.Znum, weights=self.Xnum)