MGXS.get_xs(...) routine now ensures that return value is at least a 1D numpy array

This commit is contained in:
Will Boyd 2015-10-15 16:06:32 -04:00
parent f35a68a2c7
commit 1aaf24bc4d

View file

@ -668,6 +668,7 @@ class MGXS(object):
# Eliminate trivial dimensions
xs = np.squeeze(xs)
xs = np.atleast_1d(xs)
return xs
@ -1790,6 +1791,7 @@ class ScatterMatrixXS(MGXS):
# Eliminate trivial dimensions
xs = np.squeeze(xs)
xs = np.atleast_2d(xs)
return xs
@ -2121,6 +2123,7 @@ class Chi(MGXS):
# Eliminate trivial dimensions
xs = np.squeeze(xs)
xs = np.atleast_1d(xs)
xs = np.nan_to_num(xs)
return xs