No need for redundant NaN -> zero (@samuelshaner beat me to it)

This commit is contained in:
Paul Romano 2016-08-17 10:50:33 -05:00
parent 67c2e4c3d2
commit 079f61cd1c

View file

@ -815,7 +815,6 @@ class MGXS(object):
densities = self.get_nuclide_densities('sum')
if value == 'mean' or value == 'std_dev':
xs /= densities[np.newaxis, :, np.newaxis]
xs[np.isnan(xs)] = 0.0
# Eliminate the trivial score dimension
xs = np.squeeze(xs, axis=len(xs.shape) - 1)
@ -1818,7 +1817,6 @@ class MatrixMGXS(MGXS):
densities = self.get_nuclide_densities('sum')
if value == 'mean' or value == 'std_dev':
xs /= densities[np.newaxis, :, np.newaxis]
xs[np.isnan(xs)] = 0.0
# Eliminate the trivial score dimension
xs = np.squeeze(xs, axis=len(xs.shape) - 1)
@ -3655,7 +3653,6 @@ class ScatterMatrixXS(MatrixMGXS):
densities = self.get_nuclide_densities('sum')
if value == 'mean' or value == 'std_dev':
xs /= densities[np.newaxis, :, np.newaxis]
xs[np.isnan(xs)] = 0.0
# Convert and nans to zero
xs = np.nan_to_num(xs)