From 7f0d6ff68fb888e790a97821dd9f962a78ccb7fc Mon Sep 17 00:00:00 2001 From: Daniel <92740330+PhysicsDan@users.noreply.github.com> Date: Sun, 12 Mar 2023 15:45:08 +0000 Subject: [PATCH] Update opg_tops.py Ensures `self.Znum` and `self.Xnum` are `iterable` when using a single species in the TOPS mixture --- opacplot2/opg_tops.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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)